洛谷 1092 虫食算 （dfs+剪枝）

题目描述

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

43#9865#045
+ 8468#6633
44445509678

其中#号代表被虫子啃掉的数字。根据算式，我们很容易判断：第一行的两个数字分别是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个不同的字母分别代表的数字，使得该加法算式成立。输入数据保证有且仅有一组解

输入输出格式

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

输出格式：
包含一行。在这一行中，应当包含唯一的那组解。解是这样表示的：输出N个数字，分别表示A，B，C……所代表的数字，相邻的两个数字用一个空格隔开，不能有多余的空格。

输入输出样例

输入样例#1：
5
ABCED
BDACE
EBBAA
输出样例#1：
1 0 3 4 2

说明

对于30％的数据，保证有N<＝10；

对于50％的数据，保证有N<＝15；

对于全部的数据，保证有N<＝26。

dfs+剪枝

刚开始搜时一定要从右往左搜，不然会T，虽然看起来没有什么道理，但正解都是这么搜的QAQ。。

代码

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;

int N,f,m,cnt=-1;
char A[30],B[30],C[30];
int a[30],num[4][30],used[30],pos[30],Unsigned[30];

bool print()
{
    int b,k=0;
    for(int i=N;i>=1;--i)
    {
        b=a[num[1][i]]+a[num[2][i]]+k,k=b/N,b%=N;   //判断答案是否正确 
        if(a[num[3][i]]!=b) return false;
    }
    return true;
}
void Done(int n)
{
     if(f) return ;
     if(a[num[1][0]]+a[num[2][0]]>=N) return ;  // 最高位不进位 
     if(n==N) 
       if(print())
       {
          for(int i=1;i<=N;++i) 
             cout<<a[i]<<" ";
           f=1; return ;
       }
       else return ;
      for(int i=1;i<=N;++i)  
      {
          m=1;
          if(i==N) m=0;
          if(a[num[1][i]]!=-1&&a[num[2][i]]!=-1&&a[num[3][i]]!=-1)   //判断已经搜到的答案的合法性  重要的剪枝1 ！！！ 
             if(a[num[3][i]]!=(a[num[1][i]]+a[num[2][i]])%N&&a[num[3][i]]!=((a[num[1][i]]+a[num[2][i]]+m)%N))
              return ;
          if(a[num[1][i]]!=-1&&a[num[3][i]]!=-1&&a[num[2][i]]==-1)
             if(used[(a[num[3][i]]-a[num[1][i]]+N)%N]&&used[(a[num[3][i]]-a[num[1][i]]-1+N)%N])
               return ;
          if(a[num[2][i]]!=-1&&a[num[3][i]]!=-1&&a[num[1][i]]==-1)
             if(used[(a[num[3][i]]-a[num[2][i]]+N)%N]&&used[(a[num[3][i]]-a[num[2][i]]-1+N)%N])
               return ;
          if(a[num[1][i]]!=-1&&a[num[2][i]]!=-1&&a[num[3][i]]==-1)
             if(used[(a[num[1][i]]+a[num[2][i]])%N]&&used[(a[num[1][i]]+a[num[2][i]]+m)%N])
               return ;

      } 
     for(int i=N-1;i>=0;--i)
     {
       if(!used[i]&&!f) //不重复放置 
       {
           a[pos[n]+1]=i;
           used[i]=1;
           Done(n+1);
           a[pos[n]+1]=-1;
           used[i]=0;
       }
     }
}
int main()
{
    scanf("%d",&N);
    memset(a,-1,sizeof(a));
    cin>>A>>B>>C;
    for(int j=N-1;j>=0;--j)
    {
        num[1][j+1]=A[j]-'A'+1,num[2][j+1]=B[j]-'A'+1,num[3][j+1]=C[j]-'A'+1;  // 按照字母的出现顺序枚举            
        if(!Unsigned[A[j]-'A'])//从右往左，从上往下，答案的递减可能性最大，可以优先剪去一些不必要的废枝，  重要的剪枝2 ！！！ 
        {
            Unsigned[A[j]-'A']=1;
            pos[++cnt]=A[j]-'A';
        }
        if(!Unsigned[B[j]-'A'])
        {
            Unsigned[B[j]-'A']=1;
            pos[++cnt]=B[j]-'A';
        }
        if(!Unsigned[C[j]-'A'])
        {
            Unsigned[C[j]-'A']=1;
            pos[++cnt]=C[j]-'A';
        }       
    }
    Done(0);
    return 0;
}