Luogu P1092 Insect Food Calculation
Tags: search, number theory, math
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int n,c[4][30],v[30],a[30];
void dfs(int x,int y,int t){
if (a[c[1][n]]>=0&&a[c[2][n]]>=0&&a[c[3][n]]>=0){
if((a[c[1][n]]+a[c[2][n]])>=n)
return;
}
for(int i=n-1;i>=1;i--){ //剪枝
if(a[c[1][i]]>=0&&a[c[2][i]]>=0&&a[c[3][i]]>=0){
if(((a[c[1][i]]+a[c[2][i]])%n!=a[c[3][i]])&&((a[c[1][i]]+a[c[2][i]]+1)%n!=a[c[3][i]]))
return;
}
}
if(y>n){
if(t==0)
for(int i=0;i<n;i++){
cout<<a[i]<<" ";
}
return;
}
int z=c[x][y];
if(x<=2){
if(a[z]!=-1){
dfs(x+1,y,t);
}
else{
for(int i=0;i<=n;i++){
if(!v[i]){
v[i]=1;
a[z]=i;
dfs(x+1,y,t);
v[i]=0;
a[z]=-1;
}
}
}
}
else{
t+=a[c[1][y]]+a[c[2][y]];
if(a[z]==t%n){
dfs(1,y+1,t/n);
}
if(a[z]==-1&&v[t%n]==0){
a[z]=t%n;
v[t%n]=1;
dfs(1,y+1,t/n);
a[z]=-1;
v[t%n]=0;
}
}
}
int main()
{
memset(a,-1,sizeof(a));
cin>>n;
for(int i=1;i<=3;i++){
for(int j=n;j>=1;j--){
char t;
cin>>t;
c[i][j]=t-'A';
}
}
dfs(1,1,0);
return 0;
}