184. 虫食算
所谓虫食算,就是原先的算式中有一部分被虫子啃掉了,需要我们根据剩下的数字来判定被啃掉的字母。
来看一个简单的例子:
43#9865#045
+ 8468#6633
--------------
44445506978
其中 # 号代表被虫子啃掉的数字。
根据算式,我们很容易判断:第一行的两个数字分别是 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 个不同的字母分别代表的数字,使得该加法算式成立。
输入数据保证有且仅有一组解。
输入格式
输入包含 4 行。
第一行有一个正整数 N(N≤26),后面的 3 行每行有一个由大写字母组成的字符串,分别代表两个加数以及和。
这 3 个字符串左右两端都没有空格,并且恰好有 N 位。
输出格式
输出包含一行。
在这一行中,应当包含唯一的那组解。
解是这样表示的:输出 N 个数字,分别表示 A,B,C…… 所代表的数字,相邻的两个数字用一个空格隔开,不能有多余的空格。
输入样例:
5
ABCED
BDACE
EBBAA
输出样例:
1 0 3 4 2
#include <bits/stdc++.h>
using namespace std;
const int N=30;
int n;
char e[3][N];//输入三行字母
int q[N],path[N];//q数组表示遍历字母的顺序,因为是手动模拟从右到左,所以字母不一定按顺序,要遍历先出现的字母
//path数组存最终A~n个字母分别是哪个数字
bool st[N];//是一个标志变量
bool check(){
//check函数检查某种数字的方式是否正确
for(int i=n-1,t=0;i>=0;i--){//从低位向高位遍历
//a,b,c表示一列的三个字母
int a=e[0][i]-'A',b=e[1][i]-'A',c=e[2][i]-'A';//验证a+b=c
if(path[a]!=-1&&path[b]!=-1&&path[c]!=-1){//如果三个字母都赋值了
a=path[a],b=path[b],c=path[c];//此时a,b,c是三个字母所代表的数字
if(t!=-1){//如果进位已经确定
if((a+b+t)%n!=c) return false;
if(!i&&a+b+t>=n) return false;//验证高位是否有进位,不允许有进位
t=(a+b+t)/n;//进位
}else{//进位无法确定,那么进位可能是0或1
if((a+b+0)%n!=c&&(a+b+1)%n!=c) return false;
if(!i&&a+b>=n) return false;
}
}
else t=-1; //如果有字母未赋值,就不需要再判断这一列,并将t标记为-1,表示下一次计算时的进位不确定
}
return true;
}
bool dfs(int u){
if(u==n) return true;//如果字母个数==n
for(int i=0;i<n;i++){//每个字母可以是从0~n数字中的一个
if(!st[i]){//st数组表示这个数字有没有被用过
st[i]=true;
path[q[u]]=i;//将数字i赋值给字母q[u]
if(check()&&dfs(u+1)) return true;
st[i]=false;
path[q[u]]=-1;
}
}
return false;
}
int main() {
scanf("%d",&n);
for(int i=0;i<3;i++) scanf("%s",e[i]);//输入
for(int i=n-1,k=0;i>=0;i--){
for(int j=0;j<3;j++){
int t=e[j][i]-'A';//循环所有字母,将字母以数字的形式存起来
if(!st[t]){//如果该字母未出现过,就入数组
st[t]=true;//并将它标记为已经入了数组
q[k++]=t;
}
}
}
memset(st,0,sizeof st);//将标志变量重置,以后还要用
memset(path,-1,sizeof path);//将path所有值标记为-1,表示没有字母被赋过值,因为字母的值是0~n
dfs(0);//从第一个字母开始遍历
//输出所有字母的值
for(int i=0;i<n;i++) printf("%d ",path[i]);
return 0;
}
思路
dfs依次枚举每个字母,每个字母可以选择0~n这些数。
手动做加法竖式时按照从右到左的顺序,因此我们也按照从右到左出现的次序枚举每个字母的选法,这样可以方便后续的剪枝:
从后往前枚举每一列,记当前这列的被加数、加数、和分别为 a,b,c,如果右边所有数均以确定,则当前这位的进位也是确定的,记为 t,则如果 a+b+t≠c,那么当前方案不合法;
如果右边存在某个未知数尚未确定,则上一位的进位 t 可能取0或1,那么如果 a+b≠c,a+b+1≠c 同时成立,则说明当前方案不合法;
另外对于最高位,判断是否有进位,如果有进位,则说明不合法(因为和也是 n 位数)