题解 P1092 【虫食算】

· · 题解

[NOIP2004]虫食算

总时间限制:

1000ms

内存限制:

65536kB

题目

描述

所谓虫食算,就是原先的算式中有一部分被虫子啃掉了,需要我们根据剩下的数字来判定被啃掉的字母。来看一个简单的例子:
       43#98650#45
+       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个不同的字母分别代表的数字,使得该加法算式成立。输入数据保证有且仅有一组解。

输入 输入文件alpha.in包含4行。第一行有一个正整数N(N<=26),后面的3行每行有一个由大写字母组成的字符串,分别代表两个加数以及和。这3个字符串左右两端都没有空格,从高位到低位,并且恰好有N位。

输出 输出文件alpha.out包含一行。在这一行中,应当包含唯一的那组解。解是这样表示的:输出N个数字,分别表示A,B,C……所代表的数字,相邻的两个数字用一个空格隔开,不能有多余的空格。 样例输入

5
ABCED
BDACE
EBBAA

样例输出

1 0 3 4 2

【数据规模】

对于30%的数据,保证有N≤10;
对于50%的数据,保证有N≤15;
对于全部的数据,保证有N≤26。

题目解析

刚看此题时,第一想法是暴力枚举,即用全排列一个一个填数。而对应的时间复杂度是O(n!) 而最大的n可以取到26,所以必须剪枝,不然1000ms绝对超时 先看题目的举例 ...BADC

而在填等式时,若从小的数开始倒着填,那么最高位一定是先填最大的数,但若是这样,最高位必定会进位,而题目说得清楚,都是相同的位数,所以若是这样,也要在填完所有数之后才能发现最高位进位的填数错误,这样和正着填有什么区别。 因此,倒着填时,应从大的数字开始填起,这样才能最大化剪枝 剪枝要点二 填数时从大的数字倒着往前填

若在填完数字后,再用一次高精度进行运算,那么有那么多种填数情况,肯定会超时。 其实,这里只需储存下上一位的进位,然后一位一位地进行检验即可,这样便少了数组的转换 剪枝要点三 填完数后只需按位相加,与和数进行检测

代码实现

#include<cstdio>
#include<cstring>
#include<cstdlib>
#define MAXN 100
char a[MAXN],b[MAXN],c[MAXN];
bool used[MAXN];
//在ready()预处理函数中,记录这个字母是否在之前的位数中出现过
char Rlist[MAXN];
//剪枝要点一中的字母出现顺序表
int num[MAXN];
//以字母ASCII码作为下标,表示该字符的数值
int tot;
//记录Rlist数组的下标
int n;
//n进制
void ready()
//剪枝一 预处理函数
{
    for(int i=n-1;i>=0;i--)
    //从末尾开始,以a,b,c为序,遍历储存出现顺序
    {
        if(!used[a[i]])
        //判断Rlist储存过a[i]没有
        {
            Rlist[++tot]=a[i];
            //若没有,下标++,并存进Rlist
            used[a[i]]=1;
            //标记储存过a[i]
        }//之后的b,c都是如此
        if(!used[b[i]])
        {
            Rlist[++tot]=b[i];
            used[b[i]]=1;
        }
        if(!used[c[i]])
        {
            Rlist[++tot]=c[i];
            used[c[i]]=1;
        }
    }
    memset(used,0,sizeof(used));
    //重置数组,方便在dfs中使用
    memset(num,-1,sizeof(num));
    //将储存数值的数组置为-1,-1表示没有被赋值,即不知道值
    return;
}
bool check()
//检测填的数字是否合适
{
    for(int i=n-1;i>=0;i--)
    {//因为是在填数途中检测,所以分为四种情况
        if(num[a[i]]!=-1 && num[b[i]]!=-1 && num[c[i]]!=-1)
        //情况一 这一位两个加数与和都被填过
        {
            //这时判断两个加数是不是等于和数 注意此处不知道是否有进位
            //所以进位与不进位两种情况都要判断
            //如果两种情况都不等于和 说明填数错误
            //便返回0
            if((num[a[i]]+num[b[i]])%n!=num[c[i]] && (num[a[i]]+num[b[i]]+1)%n!=num[c[i]])
                return 0;
        }
        if(num[a[i]]!=-1 && num[b[i]]!=-1 && num[c[i]]==-1)
        //情况二 知道两个加数而不知道和
        {
            //这时也有两种情况
            //进位与不进位
            //将两种情况出现时c分别的值算出来
            //如果这两种情况c的值都已经被别的字母使用过
            //说明填数错误 便返回0
            int c1=(num[a[i]]+num[b[i]])%n;
            int c2=(num[a[i]]+num[b[i]]+1)%n;
            if(used[c1] && used[c2])
                return 0;
        }
        if(num[a[i]]!=-1 && num[b[i]]==-1 && num[c[i]]!=-1)
        //情况三 知道其中的a加数与和
        {
            //求出b在进位与不进位时的两个值
            //注意 这里加n是为了防止出现负数的情况 可以理解为做减法时的退位
            int b1=(num[c[i]]-num[a[i]]+n)%n;
            int b2=(num[c[i]]-num[a[i]]-1+n)%n;
            //若两个值都被占用 说明填数错误 返回0
            if(used[b1] && used[b2])
                return 0;
        }
        if(num[a[i]]==-1 && num[b[i]]!=-1 && num[c[i]]!=-1)
        //情况四 知道其中的b加数与和
        {
            //求出a在进位与不进位时的两种值
            //注意 这里加n是为了防止出现负数的情况 可以理解为做减法时的退位
            int a1=(num[c[i]]-num[b[i]]+n)%n;
            int a2=(num[c[i]]-num[b[i]]-1+n)%n;
            //若两个值都被占用 说明填数错误 返回0
            if(used[a1] && used[a2])
                return 0;
        }
    }
    //若都可以,说明填数正确 返回1
    return 1;
}
bool OK()
//检测最终的填数是否正确
{
    int x=0;
    //进位
    for(int i=n-1;i>=0;i--)
    //剪枝三 不用高精度,只需按位计算
    {
        //从个位开始 一位一位挨着判断此位两加数相加
        //其和是否等于和数 注意要加上进位的数
        if((x+num[a[i]]+num[b[i]])%n!=num[c[i]])
        //若不等于 说明填数错误 返回0
            return 0;
        //处理此位的进位
        x=(x+num[a[i]]+num[b[i]])/n;
    }
    //若最后还有进位 说明最终结果不止n位 与题意不符 返回0
    if(x)
        return 0;
    //若成立 说明等式正确 返回1
    return 1;
}
void print()
//输出最终结果
{
    for(int i=65;i<n-1+65;i++)
    //因为使用num数组时,是用ASCII码作为下标 所以应从65开始
            printf("%d ",num[i]);
    //注意 题目要求最后没有空格
    printf("%d",num[n-1+65]);
}
void dfs(int x)
//x控制Rlist的下标,表示正在填Rlist[i]的值
{
    if(x==n+1)
    //如果填数完成
    {
        if(OK())
        //检测填数是否正确
        {
            print();
            //输出并结束程序,如果不结束,程序会寻找下一种情况
            //所以此处直接用exit(0),在子函数中结束整个程序
            exit(0);
        }
        return;
        //如果不正确,回溯回去,继续寻找正确填法
    }
    else
    //如果填数没有完成
    {
        for(int i=n-1;i>=0;i--)
        //剪枝二 从大数往小数填
        //用for循环寻找没有被用过的数字 将其填给Rlist[x]
        //然后再看看行不行,若可以,就继续dfs
        //不可以,便寻找下一个值
        //此处即用for循环一个一个找,可以填给Rlist[x]的数字
            if(!used[i])
            //判断数字i是否给其他字母使用过
            {
                num[Rlist[x]]=i;
                //将数字i赋值给Rlist[x]
                used[i]=1;
                //记录数字i被使用过
                if(!check())
                //剪枝一 每填完一位就判断一下,填进去的数是否合适
                //若不合适,就不用继续dfs了
                {//直接还原,再continue进入下一次循环
                    num[Rlist[x]]=-1;
                    used[i]=0;
                    continue;
                }
                dfs(x+1);
                //进入深搜
                used[i]=0;
                num[Rlist[x]]=-1;
                //还原现场
            }
    }
}
int main()
{
    scanf("%d\n",&n);
    //输入n进制
    gets(a);
    gets(b);
    gets(c);
    //输入a,b,c
    ready();
    //调用预处理函数
    dfs(1);
    //深搜填数
    return 0;
}