DFS

大雪菜的课（笔记）

搜索与图论（一）

1.DFS

(1).回溯

AcWing842. 排列数字

给定一个整数n，将数字1~n排成一排，将会有很多种排列方法。

现在，请你按照字典序将所有的排列方法输出。

输入格式
共一行，包含一个整数n。

输出格式
按字典序输出所有排列方案，每个方案占一行。

数据范围
1≤n≤7
输入样例：
3
输出样例：
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

#include <iostream>
using namespace std;
const int N=10;
int path[N],n;
bool p[N];
void dfs(int u){
    if(u==n){
        for(int i=0;i<n;i++)    printf("%d ",path[i]);
        puts("");
        return;
    }
    for(int i=1;i<=n;i++)
        if(!p[i]){
            path[u]=i;
            p[i]=true;
            dfs(u+1);
            p[i]=false;
        }
}
int main()
{
    scanf("%d",&n);
    dfs(0);
    return 0;
}

#include <iostream>
using namespace std;
const int N=100010;
int path[N],n;
void dfs(int u,int state){
    if(u==n){
        for(int i=0;i<n;i++)    cout<<path[i]<<' ';
        cout<<endl;
        return;
    }
    for(int i=0;i<n;i++){
        if(!(state>>i&1)){
            path[u]=i+1;
            dfs(u+1,state+(1<<i));
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    dfs(0,0);
    return 0;
}

(2).剪枝

AcWing843. n-皇后问题

n-皇后问题是指将 n 个皇后放在 n∗n 的国际象棋棋盘上，使得皇后不能相互攻击到，即任意两个皇后都不能处于同一行、同一列或同一斜线上。

现在给定整数n，请你输出所有的满足条件的棋子摆法。

输入格式
共一行，包含整数n。

输出格式
每个解决方案占n行，每行输出一个长度为n的字符串，用来表示完整的棋盘状态。

其中”.”表示某一个位置的方格状态为空，”Q”表示某一个位置的方格上摆着皇后。

每个方案输出完成后，输出一个空行。

输出方案的顺序任意，只要不重复且没有遗漏即可。

数据范围
1≤n≤9
输入样例：
4
输出样例：
.Q…
…Q
Q…
…Q.

…Q.
Q…
…Q
.Q…

#include <iostream>
using namespace std;
const int N=100;
char q[N][N];
bool row[N],col[N],dg[N],udg[N];
int n;
void dfs(int x,int y,int s){
    if(y==n)    y=0,x++;
    if(x==n){
        if(s==n){
            for(int i=0;i<n;i++)    puts(q[i]);
            puts("");
        }
        return;
    }
    q[x][y]='.';
    dfs(x,y+1,s);
    if(!row[x]&&!col[y]&&!dg[x+y]&&!udg[x-y+n]){
        q[x][y]='Q';
        row[x]=col[y]=dg[x+y]=udg[x-y+n]=true;
        dfs(x,y+1,s+1);
        q[x][y]='.';
        row[x]=col[y]=dg[x+y]=udg[x-y+n]=false;
    }
}
int main()
{
    scanf("%d",&n);
    dfs(0,0,0);
    return 0;
}

#include <iostream>
using namespace std;
const int N=100;
char q[N][N];
bool col[N],dg[2*N],udg[2*N];
int n;
void dfs(int u){
    if(u==n){
        for(int i=0;i<n;i++)    puts(q[i]);
        puts("");
    }
    for(int i=0;i<n;i++)
        if(!col[i]&&!dg[i-u+n]&&!udg[i+u]){
            q[u][i]='Q';
            col[i]=dg[i-u+n]=udg[i+u]=true;
            dfs(u+1);
            q[u][i]='.';
            col[i]=dg[i-u+n]=udg[i+u]=false;
        }
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            q[i][j]='.';
    dfs(0);
    return 0;
}