【题目链接】
https://www.luogu.org/problemnew/show/P1092
【算法】
搜索 + 剪枝
直接搜索显然会超时,考虑剪枝
1 : 优化搜索顺序
2 : 假设我们已经确定了一组(Ai,Bi,Ci),那么,如果(Ai + Bi) mod n 和(Ai + Bi + 1) mod n都不等于Ci,可以剪枝
【代码】
#include<bits/stdc++.h>
using namespace std;int i,n,len;
int a[],ord[];
char A[],B[],C[];
bool used[],vis[];
bool solved;inline bool ok()
{
int i;
for (i = n; i >= ; i--)
{
if (a[A[i]-'A'+] == -) continue;
if (a[B[i]-'A'+] == -) continue;
if (a[C[i]-'A'+] == -) continue;
if ((a[A[i]-'A'+] + a[B[i]-'A'+]) % n == a[C[i]-'A'+]) continue;
if ((a[A[i]-'A'+] + a[B[i]-'A'+] + ) % n == a[C[i]-'A'+]) continue;
return false;
}
return true;
}
inline void print()
{
int i;
for (i = ; i <= n; i++) printf("%d ",a[i]);
printf("\n");
}
inline bool check()
{
int t,b = ;
for (i = n; i >= ; i--)
{
t = (a[A[i]-'A'+] + a[B[i]-'A'+] + b) % n;
if (a[A[i]-'A'+] + a[B[i]-'A'+] + b >= n) b = ;
else b = ;
if (a[C[i]-'A'+] == t) continue;
else return false;
}
if (b) return false;
else return true;
}
inline void dfs(int dep)
{
int i;
if (dep > n)
{
if (check())
{
solved = true;
print();
}
return;
}
if (!ok()) return;
for (i = n - ; i >= ; i--)
{
if (!used[i])
{
a[ord[dep]] = i;
used[i] = true;
dfs(dep+);
if (solved) return;
used[i] = false;
a[ord[dep]] = -;
}
}
}int main()
{ memset(a,,sizeof(a));
scanf("%d%s%s%s",&n,A+,B+,C+);
for (i = n; i >= ; i--)
{
if (!vis[A[i]-'A'+])
{
ord[++len] = A[i] - 'A' + ;
vis[A[i]-'A'+] = true;
}
if (!vis[B[i]-'A'+])
{
ord[++len] = B[i] - 'A' + ;
vis[B[i]-'A'+] = true;
}
if (!vis[C[i]-'A'+])
{
ord[++len] = C[i] - 'A' + ;
vis[C[i]-'A'+] = true;
}
}
dfs(); return ;}