AcWing184. 虫食算

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 位数）