D - 哗啦啦族的加法计算
Time Limit:
12000/6000MS (Java/Others)
Memory Limit:
128000/64000KB (Java/Others)
Problem Description
小彭玉今天开始学习加法啦!
唐老师写了个加法等式,小彭玉扑通扑通就凑出来了一种合法解
唐老师又写了个加法等式,然后想问你,这个加法等式究竟有多少种可行解呢?
Input
输入包含多组数据。
每组数据第一行一个整数n,表示有n个字符串 3 ≤ n ≤ 10
接下来n行,每行有1个最多只含10个大写字母的字符串,前 n - 1 行的字符串表示加数,第 n 行的字符串表示和
每个样例最多只有26个互不相同的大写字母,每个字母表示 0 - 9 中的一个数,相同字母表示相同的数,不同字母表示不同的数输入包含多组数据。
Output
对于每组数据输出一个整数,表示不同的可行解的数量。
对于两个可行解,只要有一个字母表示不同的数字,我们就认为这两个可行解不同
Sample Input
4 TAI SHEN LA ACER
Sample Output
76
Hint
如果各个字符串长度不等,右对齐后再运算
每个字符串最左边的字母表示的数字不能为0
不保证最后一个字符串的长度大于等于前面的表示加数的字符串长度
题目大意:给出一个字母的加法等式,每个字母代表不同的数字0~9,求合法解的个数。
分析:爆搜加剪枝。从低位向高位搜索。当搜完一位后,判断是否合法,合法就搜下一位。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
char s[15][15];
int len[15];
int vis[15];
int mp[30]; //字母对应的数字
int n, ans, mx;
bool check() {
for(int i = 1; i <= n; i++)
if(mp[s[i][len[i]-1]-'A'] == 0) //判断是否出现前导0
return false;
return true;
}
void dfs(int x, int y, int sum) {
int ch = s[x][y]-'A';
if(x == n) {
if(y == mx-1) {
if(sum/10 == 0) {
if(mp[ch] == -1 && !vis[sum%10] && check())
ans++;
else if(mp[ch] == sum%10 && check())
ans++;
}
}
else if(mp[ch] != -1) {
if(sum%10 == mp[ch])
dfs(1, y+1, sum/10);
}
else if(!vis[sum%10]){
mp[ch] = sum%10;
vis[sum%10] = 1;
dfs(1, y+1, sum/10);
mp[ch] = -1;
vis[sum%10] = 0;
}
}
else {
if(ch+'A' == 0) dfs(x+1, y, sum); //判断当前字符是否为空,前面减去了'A',所以这里要加上
else if(mp[ch] != -1)
dfs(x+1, y, sum+mp[ch]);
else {
for(int i = 0; i < 10; i++)
if(!vis[i]){
mp[ch] = i;
vis[i] = 1;
dfs(x+1, y, sum+i);
mp[ch] = -1;
vis[i] = 0;
}
}
}
}
int main() {
while(~scanf("%d", &n)) {
mx = 0;
memset(vis, 0, sizeof(vis));
memset(mp, -1, sizeof(mp));
memset(s, 0, sizeof(s));
for(int i = 1; i <= n; i++) {
scanf("%s", s[i]);
len[i] = strlen(s[i]);
mx = max(mx, len[i]);
reverse(s[i], s[i]+len[i]); //反转字符串
}
ans = 0;
dfs(1, 0, 0);
printf("%d\n", ans);
}
return 0;
}