Luogu P1092 Insect Food Calculation

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;
}