大雪菜的课(笔记)
搜索与图论(一)
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;
}