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;
}