dfs练习题

POJ1321
棋盘问题
题意：#号位置可以放置棋子，给你一个n*n的棋盘，其中有多少种可以放置k个棋子的办法，同时要满足一个条件就是同一行同一列不能有两个棋子及以上
思路：dfs深搜，用vis数组标记各个列是否放置了棋子，board数组表示棋盘，dfs(i,j)表示到达了第i行，已经放置了j个棋子，注意回溯。具体实现看代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
#include <map>
#include<iomanip>
using namespace std;
typedef long long ll;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define per(i, a, n) for(int i = n-1; i >= a; i--)
#define INF 1ll<<60
const int maxn = 10+10;
int n, k;
ll ans = 0;
char board[maxn][maxn];
bool vis[maxn];
bool check(int x, int y){
    if(board[x][y] == '.') return 0;
    if(vis[y]) return 0;
    return 1;
}
void dfs(int dep, int step){
    if(step==k){
        ans++;
        return ;
    }
    if(dep>=n) return ;
    for(int i = 0; i < n; i++){
        if(check(dep,i)){
            vis[i] = 1;
            dfs(dep+1, step+1);
            vis[i] = 0;
        }
    }
    dfs(dep+1, step);
}
int main(){
    while(scanf("%d%d", &n, &k), n!=-1&&k!=-1){
        for(int i = 0; i < n; i++){
            scanf("%s", board[i]);
        }
        ans = 0;
        dfs(0, 0);
        printf("%lld\n", ans);
    }
}

HDU1016
题意：给你一个n，问有多少种解法可以使得相邻两个数之间的和为素数，且最后一个和第一个的和也为素数，第一个数为1，并打印出每种排列顺序。
思路：第一个为1，且最后一个和第一个的和为素数可以留到最后来判断，dfs(i)表示已经放了i个数字进去，检查是否为素数可以用代码中的一个一个的算或者预处理一下，因为n是小于20的，设一个数组长度为40的就行。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
#include <map>
#include<iomanip>
using namespace std;
typedef long long ll;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define per(i, a, n) for(int i = n-1; i >= a; i--)
#define INF 1ll<<60
const int maxn = 20+10;
int n, a[maxn], vis[maxn];
bool check(int x){
    int j = 2;
    while(x%j != 0) j++;
    if(x == j) return 1;
    return 0;
}
void print(){
    for(int i = 0; i < n; i++){
        printf("%d", a[i]);
        if(i!=n-1) printf(" ");
    }
    printf("\n");
}
void dfs(int step){
    if(step==n&&check(a[step-1]+1)){
        print();
        return ;
    }
    for(int i = 2; i <= n; i++){
        a[step] = i;
        if(check(a[step-1]+i)&&vis[i]==0){
            vis[i] = 1;
            dfs(step+1);
            vis[i] = 0;
        }
    }
}
int main(){
    int t = 1;;
    while(~scanf("%d", &n)){
        printf("Case %d:\n", t++);
        a[0] = 1;
        memset(vis, 0, sizeof(vis));
        dfs(1);
        printf("\n");
    }
    return 0;
}

POJ1154
题意：给你一个棋盘，其中每格有大写字母，问你从第一行第一列开始走，最多能走多少步，且满足条件经过的路径没有相同的字母。
思路：把经过的路径的字母记录下来，dfs一下就行了。ans表示的当前走的步数，ma表示最多走的步数，一开始的算一步

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
#include <map>
#include <iomanip>
using namespace std;
typedef long long ll;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define per(i, a, n) for(int i = n-1; i >= a; i--)
#define INF 1ll<<60
const int maxn = 20+10;
char board[maxn][maxn];
int n, m, ans = 1, ma = 1;
int dir[4][2] = {0, 1, 1, 0, -1, 0, 0, -1};
bool vis[30];
bool check(int x, int y){
    if(x<0||x>=n||y<0||y>=m||vis[board[x][y]-'A']) return 0;
    return 1;
}
void dfs(int x, int y){
    if(ans > ma) ma = ans;
    for(int i = 0; i < 4; i++){
        int ix = x+dir[i][0];
        int iy = y+dir[i][1];
        if(check(ix, iy)){
            vis[board[ix][iy]-'A'] = 1;
            ans++;
            dfs(ix, iy);
            vis[board[ix][iy]-'A'] = 0;
            ans--;
        }
    }
}
int main(){
    scanf("%d%d", &n, &m);
    for(int i = 0; i < n; i++){
        scanf("%s", board[i]);
    }
    vis[board[0][0]-'A'] = 1;
    dfs(0, 0);
    printf("%d", ma);
    return 0;
}

Codeforces1338B
题意：给你一棵树，然后给每条边赋值，使得叶子节点之间经过的路径的权值异或和为零，问你赋予这些路径的权值最少可以有多少种不同的数字，最多可以有多少种不同的数字。
思路：对于最小值如果两个叶子节点之间存在经过奇数条边的最小值即为3，否则为1，用dfs来求可得，对于最大值，初始值为n-1，即n-1条边，如果存在两个叶子节点公用一个结点则最大值减一，最后可得答案。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define per(i, a, n) for(int i = n-1; i >= a; i--)
#define INF 1ll<<60
const int maxn = 1e5+10;
vector<int> v[maxn];
int vis[maxn];
int flag =1;
void dfs(int step, int dep){
    if(v[step].size()==1&&vis[step]){
        if(dep%2==0) flag = 3; 
        return ;
    }
    vis[step] = 1;
    for(int i = 0; i < v[step].size(); i++){
        if(!vis[v[step][i]]){
            vis[v[step][i]] = 1;
            dfs(v[step][i], dep+1);
            vis[v[step][i]] = 0;
        }
    }
}
int main(){
    int n;
    scanf("%d", &n);
    for(int i = 1; i < n; i++){
        int x, y;
        scanf("%d%d", &x, &y);
        v[x].push_back(y), v[y].push_back(x);
    }
    for(int i = 1; i <= n; i++){
        if(v[i].size()==1){
            dfs(i, 1);
            break;
        }
    }
    int sum = 0;
    for(int i = 1; i <= n; i++){
        int ans = 0;
        if(v[i].size()>=2){
            for(int j = 0; j < v[i].size(); j++){
                if(v[v[i][j]].size()==1){
                    ans++;
                }
            }
        }
        if(ans>=2) sum += ans-1;
    }
    sum = n-1-sum;
    printf("%d %d\n", flag, sum);
    return 0;
}