[NOIP2004 提高组] 虫食算（含代码）

[NOIP2004 提高组] 虫食算

题目描述

所谓虫食算，就是原先的算式中有一部分被虫子啃掉了，需要我们根据剩下的数字来判定被啃掉的数字。来看一个简单的例子：

$$\begin{array}{r}\text{43\#9865\#045}\\ +\text{8468\#6633}\\ \underline{}\\ \text{44445509678}\end{array}$$

其中 # 号代表被虫子啃掉的数字。根据算式，我们很容易判断：第一行的两个数字分别是 $5$ 和 $3$，第二行的数字是 $5$。

现在，我们对问题做两个限制：

首先，我们只考虑加法的虫食算。这里的加法是 $n$ 进制加法，算式中三个数都有 $n$ 位，允许有前导的 $0$。

其次，虫子把所有的数都啃光了，我们只知道哪些数字是相同的，我们将相同的数字用相同的字母表示，不同的数字用不同的字母表示。如果这个算式是 $n$ 进制的，我们就取英文字母表的前 $n$ 个大写字母来表示这个算式中的 $0$ 到 $n-1$ 这 $n$ 个不同的数字：但是这 $n$ 个字母并不一定顺序地代表 $0$ 到 $n-1$。输入数据保证 $n$ 个字母分别至少出现一次。

$$\begin{array}{r}\text{BADC}\\ +\text{CBDA}\\ \underline{}\\ \text{DCCC}\end{array}$$

上面的算式是一个4进制的算式。很显然，我们只要让 $\text{ABCD}$ 分别代表 $0123$，便可以让这个式子成立了。你的任务是，对于给定的 $n$ 进制加法算式，求出 $n$ 个不同的字母分别代表的数字，使得该加法算式成立。输入数据保证有且仅有一组解。

输入格式

输入的第一行是一个整数 $n$，代表进制数。

第二到第四行，每行有一个由大写字母组成的字符串，分别代表两个加数以及和。这 $3$ 个字符串左右两端都没有空格，从左到右依次代表从高位到低位，并且恰好有 $n$ 位。

输出格式

输出一行 $n$ 个用空格隔开的整数，分别代表 $A,B,\dots$ 代表的数字。

样例 #1

样例输入 #1

5
ABCED
BDACE
EBBAA

样例输出 #1

1 0 3 4 2

提示

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 10$；
• 对于 $50\%$ 的数据，保证 $n\le 15$；
• 对于 $100\%$ 的数据，保证 $1\le n\le 26$。

题目来源

洛谷

题解

#include <cstdio>
#include <algorithm>
using namespace std;

#define maxn 30

int a[maxn], b[maxn], c[maxn]; // 存储三个字符串每个字符对应的数字索引
int num[maxn], Next[maxn], n, cnt; // num存储每个字母对应的数字，Next存储DFS顺序
char s1[maxn], s2[maxn], s3[maxn]; // 输入的三个字符串
bool used[maxn]; // 标记数字是否被使用

// 判断当前num数组是否满足加法条件
bool Judge() {
    for (int i = n - 1, x = 0; i >= 0; i--) {
        int A = num[a[i]], B = num[b[i]], C = num[c[i]];
        if ((A + B + x) % n != C) return false;
        x = (A + B + x) / n;
    }
    return true;
}

// 剪枝函数，如果当前情况不可能满足条件，提前返回
bool CanPrune() {
    if (num[a[0]] + num[b[0]] >= n)
        return true;
    for (int i = n - 1; i >= 0; i--) {
        int A = num[a[i]], B = num[b[i]], C = num[c[i]];
        if (A == -1 || B == -1 || C == -1) continue;
        if ((A + B) % n != C && (A + B + 1) % n != C)
            return true;
    }
    return false;
}

// 打印结果并退出程序
void Print() {
    for (int i = 0; i < n; i++)
        printf("%d ", num[i]);
    exit(0);
}

// 深度优先搜索尝试不同的数字分配
void dfs(int x) {
    if (CanPrune() == true) return;
    if (x == n) {
        if (Judge() == true) Print();
        return;
    }
    for (int i = n - 1; i >= 0; i--)
        if (used[i] == false) {
            num[Next[x]] = i;
            used[i] = true;
            dfs(x + 1);
            num[Next[x]] = -1;
            used[i] = false;
        }
    return;
}

// 将字母转换为索引
inline int id(char c) {
    return c - 'A';
}

// 确定DFS的顺序
void GetNext(int x) {
    if (used[x] == false) {
        used[x] = true;
        Next[cnt++] = x;
    }
    return;
}

int main() {
    scanf("%d", &n);
    scanf("%s%s%s", s1, s2, s3);
    for (int i = 0; i < n; i++) {
        a[i] = id(s1[i]);
        b[i] = id(s2[i]);
        c[i] = id(s3[i]);
        num[i] = -1;
    }
    for (int i = n - 1; i >= 0; i--) {
        GetNext(a[i]);
        GetNext(b[i]);
        GetNext(c[i]);
    }
    for (int i = 0; i < n; i++) used[i] = false;
    dfs(0);
    return 0;
}