dfs重温

题目传送门
感觉这是一道比较简单的dfs的题，我之前是直接遍历的但是超时了。
还是重新写一下，温习一下。

#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<string.h>
using namespace std;
const int maxn=2e6+10;
bool a[7][7];
bool vis[7];//记录，是每一次取得的字符不同
bool flag;
void dfs(int x)
{
	if(x==6)
	{
		flag=1;
		return;
	}
	for(int i=1;i<=6;i++)
	{
		if(!vis[i]&&a[x][i])
		{
			vis[i]=1;
			dfs(x+1);
			vis[i]=0;
		}
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int l;
		char b[maxn];
		flag=0;
		memset(a,0,sizeof(a));
		memset(vis,0,sizeof(vis));
		for(int i=0;i<6;i++)
		{
			scanf("%s",b);
			l=strlen(b);
			for(int j=0;j<l;j++)
			{
				if(b[j]=='h')
					a[i][1]=1;
				else if(b[j]=='a')
					a[i][2]=1;
				else if(b[j]=='r')
					a[i][3]=1;
				else if(b[j]=='b')
					a[i][4]=1;
				else if(b[j]=='i')
					a[i][5]=1;
				else if(b[j]=='n')
					a[i][6]=1;
			}
		}
		dfs(0);
		if(flag)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}

再来一道经典的dfs
N皇后问题

#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<string.h>
#include<cmath>
using namespace std;
int a[20];
int s;
int sum;
bool check(int x)
{
	for(int i=1;i<x;i++)
		if(a[i]==a[x]||abs(a[i]-a[x])==abs(i-x))
			return 0;
	return 1;
}
void dfs(int x)
{
	if(x==s)
	{
		sum++;
		return;
	}
	for(int i=1;i<=s;i++)
	{
		a[x+1]=i;
		if(check(x+1))
			dfs(x+1);
	}
}
int main()
{
	int n;
	int b[15];
	for(int i=1;i<=10;i++)
	{
		sum=0;
		s=i;
		dfs(0);
		b[i]=sum;
	}
	while(~scanf("%d",&n)&&n)
	{
		printf("%d\n",b[n]);
	}
	return 0;
}