点击跟博主一起玩(zuo)耍(si)
这道题没有什么难度,纯粹考察代码能力。对于每一个字母代表的数字,采用dfs的方法进行枚举。同时在枚举过程中判断是否存在某个数字可以直接推出来。存在矛盾时退出dfs。
但是裸dfs会超时,所以需要剪枝。
剪枝优化:
考虑到加法运算要有进位,所以采取从低位向高位搜索的策略,这是十分重要的思想。
强剪枝:对于当前位的更高位,就是还没搜过的位,如果满足以下条件就剪枝:当前位三个数都已搜出来(记第一位为a,第二位b,第三位c),(a+b)mod n<>c and(a+b+1)mod n<>c ,就剪枝。
代码如下
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
typedef long long ll;
bool vis[100];
char b[100][100];
int a[100],c[100];
int n;
void print(){
for (int i = 65;i < 65 + n; ++i) printf("%d ",a[i]);
printf("\n");
return ;
}
bool judge1(){
for (int i = n;i; --i)
if (a[b[i][1]] != -1 && a[b[i][2]] != -1 && a[b[i][3]] != -1 &&
(a[b[i][1]] + a[b[i][2]]) % n != a[b[i][3]] &&
(a[b[i][1]] + a[b[i][2]] + 1) % n != a[b[i][3]]) return 0;
return 1;
}
bool judge2(){
int k = 0;
for (int i = n;i > 1; --i){
k = a[b[i][1]] + a[b[i][2]] + k;
if (a[b[i][3]] !=k % n) return 0;
k /= n;
}
return 1;
}
void DFS(int m){
int i;
if (!judge1()) return ;
if (a[b[1][1]] + a[b[1][2]] >= n) return ;
if (m > n)
if(judge2())
{
print();
exit(0);
}
else return ;
else
for (i=n-1;i>=0;i--)
if (vis[i]) {
a[c[m]]=i;
vis[i]=0;
DFS(m+1);
a[c[m]]=-1;
vis[i]=1;
}
}
int main(){
int k = 1;
bool h[100];
memset(h,1,sizeof(h));
memset(vis,1,sizeof(vis));
memset(a,-1,sizeof(a));
scanf("%d",&n);
for (int j = 1;j < 4;j++)
for (int i = 1;i <= n; ++i)
cin>>b[i][j];
for (int i=n;i>0;i--)
for (int j=1;j<4;j++)
if (h[b[i][j]]) {
c[k++]=b[i][j];
h[b[i][j]]=0;
}
DFS(1);
return 0;
}