dfs笔记

首先从一个例题中入手：将一个数分成3份，最后输出所有方案（每个方案中的数由大到小排列）

首先不难想到暴力枚举

for(int i=1;i<=n;++i)
    for(int j=i;j<=n;++j)
        for(int k=j;k<=n;++k)
            if(i+j+k==n)cout<<i<<' '<<j<<' '<<k<<endl;

当然可以优化成两层循环

for(int i=1;i<=n;++i){
    for(int j=i;j<=n;++j){
        int k=n-i-j;
        if(k>=j)cout<<i<<' '<<j<<' '<<k<<endl;
        else break;
    }
}

但是如果将问题改成将n分成m份呢? 

//将n分成m份，且每份大于等于前一份
#include<bits/stdc++.h>
using namespace std;
int arr[100],m,n;
void dfs(int n,int i,int a){//dfs考虑每一次分法
    if(n==0 && i==m){
        for(int j=1;j<i;j++)cout<<arr[j]<<' ';
        cout<<endl;
    }
    if(i<=m){
        for(int j=a;j<=n;j++){
            arr[i]=j;
            dfs(n-j,i+1,j);
        }
    }
}
int main(){
    cin>>n>>m;
    dfs(n,1,1);
    return 0;
}

再看看将n分成小于等于m份

#include<bits/stdc++.h>
using namespace std;
int arr[100],m,n;
void dfs(int n,int i,int a){
    if(n==0){
        for(int j=1;j<i;j++)cout<<arr[j]<<' ';
        cout<<endl;
    }
    if(i<=m){
        for(int j=a;j<=n;j++){
            arr[i]=j;
            dfs(n-j,i+1,j);
        }
    }
}
int main(){
    cin>>n>>m;
    dfs(n,1,1);
    return 0;
}

---

然后可以看看一道经典dfs

#include<bits/stdc++.h>
using namespace std;
int n,arr[11],vis[11];//vis记录全排列中是否出现，arr记录排列
void dfs(int step){
    if(step==n+1){
        for(int i=1;i<step;i++)cout<<setw(5)<<arr[i];
        cout<<endl;
        return;
    }
    for(int i=1;i<=n;i++){
        if(vis[i]==0){
            vis[i]=1;
            arr[step]=i;
            dfs(step+1);
            vis[i]=0;// ！！！
        }
    }
}
int main(){
    cin>>n;
    dfs(1);
    return 0;
}

 插入一个函数：next_permutation，（用过都说好用）

int n,arr[11];
int main(){
    cin>>n;
    for(int i=1;i<11;i++)arr[i]=i;
    do{
        for(int i=1;i<=n;i++)cout<<setw(5)<<arr[i];
        cout<<endl;
    }while(next_permutation(arr+1,arr+n+1));
    return 0;
}

---

洛谷 P1036 选数

miller rabin算法是一个判断大素数的方法，去学习一下 

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,k,arr[21],ans;
ll qpow(ll x,ll n,ll q){
    ll res=1;
    while(n){
        if(n&1)res=res*x%q;
        n>>=1;
        x=x*x%q;
    }
    return res;
}
bool query(int x){//miller_rabin
    int test[]={3,5,7,13};
    if(x==1)return 0;
    int t=x-1,k=0;
    while(!(t&1))k++,t>>=1;//x=2^k*t;
    for(int i=0;i<4;i++){
        if(x==test[i])return 1;
        ll a=qpow(test[i],t,x),nxt=a;
        for(int j=1;j<=k;j++){
            nxt=a*a%x;
            if(nxt==1 && a!=1 && a!=x-1)return 0;
            a=nxt;
        }
        if(a!=1)return 0;
    }
    return 1;
}
void dfs(int i,int a,int sum){
    if(i==k && query(sum)){
        ans++;
        return ;
    }
    for(int j=a;j<n;j++) dfs(i+1,j+1,sum+arr[j]);
}
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++)cin>>arr[i];
    sort(arr,arr+n);
    dfs(0,0,0);
    cout<<ans;
    return 0;
}

---

洛谷 P1451 求细胞数量

连通块的题目，bfs，dfs都可以

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,a[110][110],cnt;
string s[110];
int dx[]={1,-1,0,0};
int dy[]={0,0,1,-1};
void dfs(int x,int y){
    a[x][y]=0;
    for(int i=0;i<4;i++){
        x+=dx[i],y+=dy[i];
        if(x>=0 && x<n && y>=0 && y<m && a[x][y]) dfs(x,y);
        x-=dx[i],y-=dy[i];
    }
}
int main(){
    cin>>n>>m;
    for(int i=0;i<n;i++){
        cin>>s[i];
        for(int j=0;j<m;j++)
            a[i][j]=s[i][j]-'0';
    }
    /*这里有个更好的做法
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
            scanf("%1d",&a[i][j]);
    */
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
            if(a[i][j]){ dfs(i,j); cnt++; }
    cout<<cnt;
    return 0;
}