1.DFS:
递归,构建搜索树。
【例1】全排列问题:
#include<cstdio>
#include<iostream>
using namespace std;
int n;
bool vis[1005];
int a[1005];
void print(){
for(int i=1;i<=n;i++) printf("%d ",a[i]);
printf("\n");
}
void DFS(int x){
if(x==n+1){
print();
return;
}
for(int i=1;i<=n;i++){
if(vis[i]) continue;
a[x]=i;
vis[i]=1;
DFS(x+1);
vis[i]=0;
}
}
int main(){
scanf("%d",&n);
DFS(1);
return 0;
}
【例2】八皇后问题(洛谷P1219):
(https://www.luogu.com.cn/problem/P1219)
对角线分别为i+j,i-j+n。数组大小要保证大于max(i+j,i-j+n)。
#include<cstdio>
#include<iostream>
using namespace std;
int n;
const int N=520;
int cnt;
int a[N],c[N],d[N],p[N];
void print(){
for(int i=1;i<=n;i++) printf("%d ",p[i]);
printf("\n");
}
void dfs(int x){
if(x==n+1){
if(cnt<=2) print();
cnt++;
return;
}
for(int i=1;i<=n;i++){
if(a[i]==0&&c[i-x+n]==0&&d[i+x]==0){
p[x]=i;
a[i]=1;
c[i-x+n]=1;
d[i+x]=1;
dfs(x+1);
a[i]=0;
c[i-x+n]=0;
d[i+x]=0;
}
}
}
int main(){
scanf("%d",&n);
dfs(1);
printf("%d\n",cnt);
return 0;
}
2.BFS:
数连通块最好用BFS:
DFS最多n*m层,会爆栈,而BFS最多n+m层。
【例1】数连通块:
#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
queue<pair<int,int> > q;
const int dx[4]={0,0,1,-1};
const int dy[4]={1,-1,0,0};
int n,m,cnt;
bool vis[105][105];
int a[105][105];
void BFS(int sx,int sy){
q.push(make_pair(sx,sy));
while(!q.empty()){
pair<int,int> f=q.front();
q.pop();
int x=f.first,y=f.second;
for(int i=0;i<4;i++){
int nx=x+dx[i];
int ny=y+dy[i];
if(nx<1||nx>n||ny<1||ny>m) continue;
if(vis[nx][ny]) continue;
if(a[nx][ny]!=a[x][y]) continue;
vis[nx][ny]=1;
q.push(make_pair(nx,ny));
}
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++) scanf("%d",&a[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
if(!vis[i][j]){
vis[i][j]=1;
cnt++;
BFS(i,j);
}
}
printf("%d\n",cnt);
return 0;
}
【例2】最短路问题:
有向/无向图,且边权为相同数,求从1到n的最短路。
#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N=10005;
queue<int> q;
int n,m,tot;
int head[N],dis[N];
struct node{
int to,next;
}edge[N<<1];
void add(int u,int v){
edge[tot].next=head[u];
edge[tot].to=v;
head[u]=tot++;
}
void BFS(int s){
q.push(s);
dis[s]=0;
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(dis[v]==-1){
dis[v]=dis[u]+1;
q.push(v);
}
}
}
}
int main(){
memset(head,-1,sizeof(head));
memset(dis,-1,sizeof(dis));
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int x,y;
scanf("%d%d",&x,&y);
add(x,y); add(y,x);
}
BFS(1);
printf("%d",dis[n]);
return 0;
}
3.记忆化搜索:
每个问题求解一次,每次用O(1)的时间看看是否求过。
【例1】数字三角形:
(1)DP:
#include<cstdio>
#include<iostream>
using namespace std;
int n;
int mp[105][105],dp[105][105];
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++) scanf("%d",&mp[i][j]);
for(int i=n;i>=1;i--)
for(int j=1;j<=i;j++) dp[i][j]=mp[i][j]+max(dp[i+1][j],dp[i+1][j+1]);
printf("%d",dp[1][1]);
return 0;
}
(2)DFS:
#include<cstdio>
#include<iostream>
using namespace std;
int n;
int mp[105][105],dp[105][105];
int DFS(int x,int y){
if(dp[x][y]>0) return dp[x][y];
if(x==n+1) return 0;
return dp[x][y]=mp[x][y]+max(DFS(x+1,y),DFS(x+1,y+1));
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++) scanf("%d",&mp[i][j]);
printf("%d",DFS(1,1));
return 0;
}
【例2】滑雪(洛谷P1434):
(https://www.luogu.com.cn/problem/P1434)
因为状态转移方程不好定义,所以直接用记忆化搜索来代替。
#include<cstdio>
#include<iostream>
using namespace std;
int n,m,ans;
int dp[1005][1005],g[1005][1005];
int dx[4]={1,-1,0,0};
int dy[4]={0,0,-1,1};
int DFS(int x,int y){
if(dp[x][y]) return dp[x][y];
dp[x][y]=1;
for(int i=0;i<4;i++){
int nx=x+dx[i];
int ny=y+dy[i];
if(nx<1||ny<1||nx>n||ny>m) continue;
if(g[x][y]>=g[nx][ny]) continue;
dp[x][y]=max(dp[x][y],DFS(nx,ny)+1);
}
return dp[x][y];
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++) scanf("%d",&g[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
dp[i][j]=DFS(i,j);
ans=max(ans,dp[i][j]);
}
printf("%d",ans);
return 0;
}