基础算法--递归搜索DFS练习总结(下)

Acwing部分练习:

1114. 棋盘问题

 DFS code:

#include<iostream>
using namespace std;
int n,k,ans;
char ch[10][10];
bool flag[10];//记录每列放没放棋子
//x表示枚举到了第几行,cnt记录当前放了几个棋子
void dfs(int x,int cnt){
    //放好棋子就计数
    if(cnt==k) {ans++;return ;}
    //判断访问行数
    if(x>n) return ;
    //枚举列
    for(int i=1;i<=n;i++){
        //选棋子
        if(ch[x][i]=='#'&&flag[i]==false){
            flag[i]=true;
            dfs(x+1,cnt+1);//dfs递归
            flag[i]=false; //回溯
        }
    }
    //强制访问下一行
    dfs(x+1,cnt);
}
int main(){
    while(cin>>n>>k){
        if(n==-1) return 0;
        for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) cin>>ch[i][j];
        ans=0;   //更新ans
        dfs(1,0);//从第一行开始放棋子
        cout<<ans<<endl;
    }
    return 0;
}

洛谷部分练习:

P1683 入门

 DFS code:

#include<iostream>
using namespace std;
int n,m,ans=1;
char ch[25][25];
int dx[4]={-1,0,1,0}; //x方向
int dy[4]={0,1,0,-1}; //y方向
bool flag[25][25];    //状态数组0表示未选,1表示被选
//当前访问的坐标为(x,y)
void dfs(int x,int y){
    for(int i=0;i<4;i++){
        int a=x+dx[i],b=y+dy[i];
        //位置不能越界并且应该为安全位置
        if(a>=1&&a<=n&&b>=1&&b<=m&&ch[a][b]=='.'){
            //选中没有被选的状态坐标
            if(flag[a][b]==false){
                flag[a][b]=true;//标记
                ans++;          //计数
                dfs(a,b);       //递归
            }
        }
    }
}
int main(){
    cin>>m>>n; for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) cin>>ch[i][j];
    for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) if(ch[i][j]=='@'){
        flag[i][j]=true; //标记开始位置
        dfs(i,j);
    }
    cout<<ans<<endl;
    return 0;
}

P1596 [USACO10OCT] Lake Counting S

  DFS code:

#include<iostream>
using namespace std;
const int N=105;
int n,m,ans;
char ch[N][N];
bool flag[N][N];
int dx[8]={-1,-1,0,1,1,1,0,-1};
int dy[8]={0,1,1,1,0,-1,-1,-1};
void dfs(int x,int y){
    for(int i=0;i<8;i++){
        int a=x+dx[i],b=y+dy[i];
        if(a>=1&&a<=n&&b>=1&&b<=m&&ch[a][b]=='W'&&flag[a][b]==false){
            flag[a][b]=true;
            dfs(a,b);
        }
    }
}
int main(){
    cin>>n>>m; for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) cin>>ch[i][j];
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(ch[i][j]=='W'&&flag[i][j]==false){
                flag[i][j]=true;
                dfs(i,j);
                ans++;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}

P1025 [NOIP2001 提高组] 数的划分

 DFS code:

#include<iostream>
using namespace std;
int n,k,a[10];
long long ans;
//x表示枚举到第几个位置,startt表示接下来的数一定不小于x,sum表示已选数之和
void dfs(int x,int startt,int sum){
    if(x>n+1||sum>n) return ;
    if(x==k+1){
        if(sum==n) ans++;
        return ;
    }
    //式子sum+i*(k-x+1)<=n用于提前剪枝,否则枚举到最后会超时
    for(int i=startt; sum+i*(k-x+1)<=n ;i++){
        a[x]=i;
        startt=i;
        dfs(x+1,startt,sum+a[x]);
    }
    
}
int main(){
    cin>>n>>k;
    dfs(1,1,0);
    cout<<ans<<endl;
    return 0;
}

DP code:

#include<iostream>
using namespace std;
int n, k, f[205][10];
int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++){
        f[i][1] = 1; f[i][0] = 1;
    }
    for (int x = 2; x <= k; x++){
        f[1][x] = 0; f[0][x] = 0;
    }
    for (int i = 2; i <= n; i++) {
        for (int x = 2; x <= k; x++) {
            if (i > x)
                f[i][x] = f[i - 1][x - 1] + f[i - x][x];
            else
                f[i][x] = f[i - 1][x - 1];
        }
    }
    cout << f[n][k];
    return 0;
}

P1019 [NOIP2000 提高组] 单词接龙

DFS code:

#include<iostream>
using namespace std;
int n,ans;
int flag[25]; //记录每个单词使用次数
int g[25][25];//g[i][j]记录第i个单词能否接到第j个单词后面,值表示重合长度
string s[25]; //存单词
void dfs(string t,int x){
    ans=max(ans,(int)t.size());
    flag[x]++;
    for(int i=0;i<n;i++){
        if(g[x][i]&&flag[i]<2){
            dfs(t+s[i].substr(g[x][i]),i);
        }
    }
    flag[x]--;
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++) cin>>s[i];
    char startt;
    cin>>startt;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            string a=s[i],b=s[j];
            for(int k=1;k<min(a.size(),b.size());k++){
                if(a.substr(a.size()-k,k)==b.substr(0,k)){
                    g[i][j]=k;
                    break;
                }
            }
        }
    }
    for(int i=0;i<n;i++){
        if(s[i][0]==startt){
            dfs(s[i],i);
        }
    }
    cout<<ans<<endl;
    return 0;
}

 晚安了各位zzz...