【洛谷P1092】【NOIP2004 tg】虫食算【DFS】

题意

给出N位加N位的N进制加法式，判断每个字母代表哪个数字。
题目链接：洛谷 虫食算

分析

先将字母转为数字，然后DFS数位，枚举能填哪个数字，然后判断能否使当前式子成立，直到最后一位。

具体地，对于式子能否成立的判断，要看有没有进位，有的话就对于$(a+b+1)$判断，否则为$(a+b)$

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#pragma GCC optimize(2)
using namespace std;

string s[4];
int n,ans[30],t[30],v[30],ff,num;

inline void output()
{
	for(int i=1;i<=n;i++) cout<<ans[i]<<' ';
	ff=1;
}

inline bool check()
{
	int add=0;
	for(int i=n;i>0;i--)
	{
		int a=ans[s[1][i-1]-64],b=ans[s[2][i-1]-64],c=ans[s[3][i-1]-64];
		if(a!=-1&&b!=-1&&c!=-1)
		{
			if(add!=-1)
			{
				if((a+b+add)%n!=c) return false;
				if(i==1&&(a+b+add)/n>0) return false;
				add=(a+b+add)/n;
			}
			else
			{
				if((a+b)%n!=c&&(a+b+1)%n!=c) return false;
				if(i==1&&(a+b)/n>0) return false;
			}
		}
		else add=-1;
	}
	return true;
}

inline void dfs(int x)
{
	if(ff) return;
	if(x>n) output();
	for(int i=0;i<n;i++)
	{
		if(!v[i])
		{
			v[i]=1;
			ans[t[x]]=i;
			if(check()) dfs(x+1);
			v[i]=0;
			ans[t[x]]=-1;
		}
	}
}

int main()
{
	scanf("%d",&n);
	cin>>s[1]>>s[2]>>s[3];
	for(int i=n-1;i>=0;i--)
	{
		for(int j=1;j<=3;j++)
		{
			if(!v[s[j][i]-64])
			{
				v[s[j][i]-64]=1;
				t[++num]=s[j][i]-64;
			}
		}
	}
	memset(v,0,sizeof(v));
	memset(ans,-1,sizeof(ans));
	dfs(1);
	return 0;
}