【ybtoj】虫食算

最新推荐文章于 2022-03-09 20:06:31 发布

原创最新推荐文章于 2022-03-09 20:06:31 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dfs #剪枝 #c++

YbtOJ 专栏收录该内容

24 篇文章

订阅专栏

虫食算

题目来源

解题思路

这道题其实和数独游戏差不多，但是它的合法条件更复杂，要符合加法运算。
同样是要剪枝

Code

#include<cstring>
#include<cstdio>
using namespace std;
int n,tot,c[30],cc[30],ans[30];
char a[30],s[5][30];
bool check()
{
	int o=0;
	for(int i=n;i>=1;i--)
	{
		int x=ans[s[1][i]-65],y=ans[s[2][i]-65],z=ans[s[3][i]-65];
		if(x!=-1&&y!=-1&&z!=-1)
		{
			if(o!=-1)
			{
				if((x+y+o)%n!=z)
				 return 0;
				if(i==1&&x+y+o>=n)
				 return 0;
				o=(x+y+o)/n;
			}
			else 
			{
				if((x+y+0)%n!=z&&(x+y+1)%n!=z)
				 return 0;
				if(i==1&&x+y>=n)
				 return 0;
			}
		}
		else 
		 o=-1;
	}
	return 1;
}
bool dfs(int x)
{
	if(x==tot+1) 
	 return 1;
	for(int i=0;i<n;i++)
	 if(cc[i]==0)
	 {
	 	ans[a[x]-65]=i;
	 	cc[i]=1;
	 	if(check()&&dfs(x+1))
		  return 1;
	 	ans[a[x]-65]=-1;
	 	cc[i]=0;
	 }
	return 0;
}
int main()
{
	memset(ans,-1,sizeof(ans));
	scanf("%d",&n);
	for(int i=1;i<=3;i++)
	 scanf("%s",s[i]+1);
	for(int j=n;j>=1;j--)
	 for(int i=1;i<=3;i++)
	  if(c[s[i][j]-65]==0)
	  {
	  	c[s[i][j]-65]=1;
	  	a[++tot]=s[i][j];
	  }
	dfs(1);
	for(int i=0;i<n;i++)
	 printf("%d ",ans[i]);  
	return 0;
}