洛谷 P1092 [NOIP2004 提高组] 虫食算

L_D.M.

于 2024-08-08 16:33:17 发布

阅读量719

点赞数 6

分类专栏： C++算法 C++不定时更新文章标签：算法 c++

本文链接：https://blog.csdn.net/xtys4574274584/article/details/141023825

版权

C++不定时更新同时被 2 个专栏收录

15 篇文章 0 订阅

订阅专栏

C++算法

10 篇文章 0 订阅

订阅专栏

洛谷 P1092 [NOIP2004 提高组] 虫食算

[NOIP2004 提高组] 虫食算

题目描述

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

$\begin{aligned} \verb!43#9865#045! \\ +\qquad \verb!8468#6633! \\[-1em]\underline{\kern{8em}} \\ \verb!44445509678! \\ \end{aligned}$

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

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

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

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

$\begin{aligned} \verb!BADC! \\ +\quad \verb!CBDA! \\[-1em]\underline{\kern{4em}} \\ \verb!DCCC! \\ \end{aligned}$

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

5
ABCED
BDACE
EBBAA

样例输出 #1

1 0 3 4 2

提示

数据规模与约定

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

原题解是高斯消元的做法，但是很遗憾的我的技术不行，看到矩阵后头都大了，就想到一种更简单的想法。

首先我们要理解 $n$ 进制加法 $n$ 进制加法就是 $逢 n 进 1$ ，当然，这道题的进位也只能是1，简单一推就推出来了.那么这道题的思路大体就是从第一位的值开始搜，一直到合法为止，我们就想到了剪枝。且最高位不能有进位。如果有进位，显然第 $3$ 个串的长度不会是 $n$ ，而是 $n + 1$ ，这并不合法。所以一次剪枝是不够的。
在这里插入图片描述
就比如这个十进制加法计算演示，我们先看个位，
$(4 + 2 + 0) m o d 10 = 6$ ,没有问题，再看十位，
$6\neq 7$ ，所以本次假设不成立，又进行下一次枚举。

那由此，我们可以推出，若 $A, B$ 两加数的第 $i$ 位与和 $(C)$ 的第 $i$ 位有下面关系，那么我们直接 $co n t in u e$ 或者 $re t u r n$ 。

$N$ 表示进制

$\neq C[i]$
$\neq C[i]$

#include <bits/stdc++.h>
using namespace std;
const int maxn = 35;
int a[maxn],b[maxn],c[maxn];
int num[maxn],Next[maxn],n,cnt;
char s1[maxn],s2[maxn],s3[maxn];
bool used[maxn];
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);
    return ;
}
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';
}
void GetNext(int x) {
    if(used[x] == false) {
        used[x] = true;
        Next[cnt ++]=x;
    }
    return;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
    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;
}