DFS搜尋排列與組合
阿新 • • 發佈:2019-01-29
練習使用DFS搜尋12以內的五位數字的排列例如
1 2 3 4 5
1 2 3 4 6
...
...
...
12 11 10 9 7
12 11 10 9 8
利用數學排列與組合方法可知結果為12*11*10*9*8 = 95040
主要就是用到一個數組記錄訪問的狀態,利用一個solve()函式儲存搜尋結果,中間有很多小坑,自己慢慢踩吧,附上本人渣程式碼。
這種也可以不用for迴圈初狀態,main函式直接一個DFS即可,更加的簡潔,但是遍歷棋盤的還是迴圈初狀態比較好。第二個是後期的簡潔版的DFS程式碼。(2017/3/25)
#include <iostream>
using namespace std;
int num[5] = { 0 }; //儲存遍歷的數字
bool idx[13] = { false }; //儲存訪問的狀態
int count = 0;
void solve()
{
for (int i = 0; i < 5; i++)
{
cout << num[i] << " ";
}
count++;
cout << endl;
}
void dfs(int i, int k)
{
if (i <= 0 || i >= 13 ) //越界條件
{
return;
}
idx[i] = true; //用過的數字狀態為true
num[k - 1] = i;
if (k == 5) //返回條件
{
solve();
return;
}
for (int x = 1; x <= 12; x++)
{
if (idx[x] == false)
{
idx[x] = true;
dfs(x, k + 1 );
idx[x] = false;
}
}
}
int main()
{
for (int i = 1; i <= 12; i++)
{
dfs(i, 1);
idx[i] = false; //這一步非常關鍵,因為這個位置除錯了兩天,開始的第一個數字用過之後一定要更改狀態
}
cout << "count: " << count << endl;
return 0;
}
//更為簡潔的DFS搜尋
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
int n, m;
int num[10];
bool idx[10];
void dfs(int n_, int m_)
{
if (m_ == m)
{
for(int i=0; i<m;i++)
cout<<num[i];
return;
}
for (int x = 0; x <= n; x++)
{
if (idx[x] == false)
{
idx[x] = true;
num[m_] = x;
dfs(x, m_ + 1);
idx[x] = false;
}
}
}
void init()
{
for (int i = 1; i <= n; i++)
idx[i] = false;
}
int main()
{
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
while (~scanf("%d%d", &n, &m) && (m || n))
{
init();
dfs(n, 0);
}
return 0;
}
搜尋組合的DFS
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
int n, m;
int number[10];
void dfs(int pos, int num)
{
if (num == m)
{
for (int i = 0; i < m; i++)
cout << number[i] << " ";
cout << endl;
}
for (int k = pos; k <= n; k++)
{
number[num] = k;
dfs(k + 1, num + 1);
}
}
void init()
{
memset(number, 0, sizeof(number));
}
int main()
{
freopen("in.txt", "r", stdin);
freopen("out.txt", "w", stdout);
while (~scanf("%d%d", &n, &m) && (m || n))
{
init();
dfs(1, 0);
}
return 0;
}