1. 程式人生 > >DFS搜尋排列與組合

DFS搜尋排列與組合

練習使用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;
}