题目描述
所谓虫食算,就是原先的算式中有一部分被虫子啃掉了,需要我们根据剩下的数字来判定被啃掉的字母。来看一个简单的例子:
43#9865#045
+ 8468#6633
44445509678
其中#号代表被虫子啃掉的数字。根据算式,我们很容易判断:第一行的两个数字分别是5和3,第二行的数字是5。
现在,我们对问题做两个限制:
首先,我们只考虑加法的虫食算。这里的加法是N进制加法,算式中三个数都有N位,允许有前导的0。
其次,虫子把所有的数都啃光了,我们只知道哪些数字是相同的,我们将相同的数字用相同的字母表示,不同的数字用不同的字母表示。如果这个算式是N进制的,我们就取英文字母表午的前N个大写字母来表示这个算式中的0到N-1这N个不同的数字:但是这N个字母并不一定顺序地代表0到N-1)。输入数据保证N个字母分别至少出现一次。
BADC
+CBDA
DCCC
上面的算式是一个4进制的算式。很显然,我们只要让ABCD分别代表0123,便可以让这个式子成立了。你的任务是,对于给定的N进制加法算式,求出N个不同的字母分别代表的数字,使得该加法算式成立。输入数据保证有且仅有一组解
输入输出格式
输入格式:
包含四行。第一行有一个正整数N(N<=26),后面的3行每行有一个由大写字母组成的字符串,分别代表两个加数以及和。这3个字符串左右两端都没有空格,从高位到低位,并且恰好有N位。
输出格式:
包含一行。在这一行中,应当包含唯一的那组解。解是这样表示的:输出N个数字,分别表示A,B,C……所代表的数字,相邻的两个数字用一个空格隔开,不能有多余的空格。
输入输出样例
输入样例#1:
5
ABCED
BDACE
EBBAA
输出样例#1:
1 0 3 4 2
说明
对于30%的数据,保证有N<=10;
对于50%的数据,保证有N<=15;
对于全部的数据,保证有N<=26。
dfs+剪枝
刚开始搜时一定要从右往左搜,不然会T,虽然看起来没有什么道理,但正解都是这么搜的QAQ。。
代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
int N,f,m,cnt=-1;
char A[30],B[30],C[30];
int a[30],num[4][30],used[30],pos[30],Unsigned[30];
bool print()
{
int b,k=0;
for(int i=N;i>=1;--i)
{
b=a[num[1][i]]+a[num[2][i]]+k,k=b/N,b%=N; //判断答案是否正确
if(a[num[3][i]]!=b) return false;
}
return true;
}
void Done(int n)
{
if(f) return ;
if(a[num[1][0]]+a[num[2][0]]>=N) return ; // 最高位不进位
if(n==N)
if(print())
{
for(int i=1;i<=N;++i)
cout<<a[i]<<" ";
f=1; return ;
}
else return ;
for(int i=1;i<=N;++i)
{
m=1;
if(i==N) m=0;
if(a[num[1][i]]!=-1&&a[num[2][i]]!=-1&&a[num[3][i]]!=-1) //判断已经搜到的答案的合法性 重要的剪枝1 !!!
if(a[num[3][i]]!=(a[num[1][i]]+a[num[2][i]])%N&&a[num[3][i]]!=((a[num[1][i]]+a[num[2][i]]+m)%N))
return ;
if(a[num[1][i]]!=-1&&a[num[3][i]]!=-1&&a[num[2][i]]==-1)
if(used[(a[num[3][i]]-a[num[1][i]]+N)%N]&&used[(a[num[3][i]]-a[num[1][i]]-1+N)%N])
return ;
if(a[num[2][i]]!=-1&&a[num[3][i]]!=-1&&a[num[1][i]]==-1)
if(used[(a[num[3][i]]-a[num[2][i]]+N)%N]&&used[(a[num[3][i]]-a[num[2][i]]-1+N)%N])
return ;
if(a[num[1][i]]!=-1&&a[num[2][i]]!=-1&&a[num[3][i]]==-1)
if(used[(a[num[1][i]]+a[num[2][i]])%N]&&used[(a[num[1][i]]+a[num[2][i]]+m)%N])
return ;
}
for(int i=N-1;i>=0;--i)
{
if(!used[i]&&!f) //不重复放置
{
a[pos[n]+1]=i;
used[i]=1;
Done(n+1);
a[pos[n]+1]=-1;
used[i]=0;
}
}
}
int main()
{
scanf("%d",&N);
memset(a,-1,sizeof(a));
cin>>A>>B>>C;
for(int j=N-1;j>=0;--j)
{
num[1][j+1]=A[j]-'A'+1,num[2][j+1]=B[j]-'A'+1,num[3][j+1]=C[j]-'A'+1; // 按照字母的出现顺序枚举
if(!Unsigned[A[j]-'A'])//从右往左,从上往下,答案的递减可能性最大,可以优先剪去一些不必要的废枝, 重要的剪枝2 !!!
{
Unsigned[A[j]-'A']=1;
pos[++cnt]=A[j]-'A';
}
if(!Unsigned[B[j]-'A'])
{
Unsigned[B[j]-'A']=1;
pos[++cnt]=B[j]-'A';
}
if(!Unsigned[C[j]-'A'])
{
Unsigned[C[j]-'A']=1;
pos[++cnt]=C[j]-'A';
}
}
Done(0);
return 0;
}