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;
}