【ybtoj 高效进阶 1.4】【深搜】 虫食算

【ybtoj 高效进阶 1.4】【深搜】 虫食算

题目

---

解题思路

记录下字母的出现顺序（从右往左）
枚举每个字母可能的数字

a是加数一，b是加数二，c是和

当这三个数已经填数时

• 如果(a+b+w)%n!=c说明不可行
  如果最高位还有进位也不可行
• 当右边还有数没填
  如果(a+b)%n!=c且(a+b+1)%n!=c，说明不可行
  如果最高位有进位不可行

否则进位赋为-1

---

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
string s[5];
char q[30];
int n,t,p[30],ans[30],use[30];
bool check()
{
    
    
     int w=0;
	 for (int i=n;i>0;i--)
	 {
    
    
	 	 
	 	 int x=ans[s[1][i-1]-64],y=ans[s[2][i-1]-64],z=ans[s[3][i-1]-64];
	 	 if (x!=-1&&y!=-1&&z!=-1)
	 	    if (w!=-1)
			{
    
    
	 	 	   if ((x+y+w)%n!=z) return false;
	 	 	   if (i==1&&x+y+w>=n) return false;
	 	 	   w=(x+y+w)/n;
		    }
		    else
			{
			   if ((x+y)%n!=z&&(x+y+1)%n!=z) return false;
			   if (i==1&&x+y>n) return false;
		    }
		    else w=-1;
	 }
	 return true;
}
bool dfs(int x)
{     
	 if (x>n) return true;
	 for (int i=0;i<n;i++)
	     if (!use[i])
	     { 
	     	ans[q[x]-64]=i;
	     	use[i]=1; 
	     	if (check()&&dfs(x+1))
	     	   return true;
	     	use[i]=0; 
	     	ans[q[x]-64]=-1;
		 } 
	 return false;
}
int main()
{
    
    
	memset(ans,-1,sizeof(ans));
	scanf("%d",&n);
    cin>>s[1];
    cin>>s[2];
    cin>>s[3];
	for (int i=n;i>0;i--)
	    for (int j=1;j<=3;j++)
	        if (!p[s[j][i-1]-65])
	        {
    
    
	        	p[s[j][i-1]-65]=1;
	        	q[++t]=s[j][i-1]; 
			}
	dfs(1);
	for (int i=1;i<=n;i++)
	    printf("%d ",ans[i]);
	return 0; 
}