L3-015. 球队“食物链”-PAT

dfs和简单状压

思路其实就是简单的dfs后加一个状压作为小小的剪枝吧。

代码如下:

#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
char mat[25][25];
int a[25], vis[25], f, truevis[30][1050000], book[25][25];//f为标记变量（是否存在食物链）。
void dfs(int x, int step, int n, int cur)//cur用二进制表示当前状态
{
    if(f) return;
    a[step] = x;
    if(step == n-1)//已经选够了n个球队。
    {
        if(book[x][1])//判断一下是否能形成环。
        {
            f = 1;
            printf("%d", a[0]);
            for(int i = 1; i < n; i++)
                printf(" %d", a[i]);
            printf("\n");
        }
        return ;
    }
    for(int i = 1; i <= n; i++)
    {
        int nex = cur | (1<<i);
        if(vis[i] == 0 && book[x][i] && !truevis[i][cur])//truevis[i][cur]当前的状态为cur,下一个选i。
        {
            vis[i] = 1;
            dfs(i, step+1, n, nex);
            if(f) return;
            vis[i] = 0;
            truevis[i][cur] = 1;//如果没有形成食物链，那么标记此状态。
        }
    }
    return ;
}
int main()
{
    int n;
    scanf("%d", &n);
    memset(vis, 0, sizeof(vis));
    memset(truevis, 0, sizeof(truevis));
    memset(a, 0, sizeof(a));
    memset(book, 0, sizeof(book));
    for(int i = 1; i <= n; i++)
        scanf("%s", mat[i]+1);
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            if(mat[i][j] == 'L')
                book[j][i] = 1;
            if(mat[i][j] == 'W')
                book[i][j] = 1;
        }
    }
    f = 0;
    vis[1] = 1;
    dfs(1, 0, n, 1);//由于形成环，若存在，必定从1开始字典序最小
    if(!f)  printf("No Solution\n");
    return 0;
}