题目地址:http://codevs.cn/problem/1064/
分析:
裸搜不能ac。(强烈膜拜前面的高能剪枝。)
所以我们要设法将复杂度中的阶乘降低到指数,其可行性是由于当我们全部知道某位是否进位时,就可以唯一的确定一组答案。
所以我们可以枚举进位个数 然后每次设法把答案解出来 判断可行性(一得是整数 二不能出界0..n-1 三不能重复)。
不过如果每枚举一种进位就解一次方程组,时间会达到O(2^n * n^3),比较慢,所以我们应该注意到进位只对方程组的常数项有影响 所以我们直接设常数项d1...dn,只解一次方程组,把常数项系数搞出来,每次枚举的时候直接代值。
细节见代码。数组equ的每一行的前26项是未知数系数,后26项是常数项系数。
代码:
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#define ZERO 1e-6
inline bool equal(double a, double b) { double dif = a - b; return (dif > -ZERO && dif < ZERO); }
double equ[26][52];
int arg[26], var[26];
bool filt[26];
inline void dogauss(int n) {
for (int unk = 0; unk < n; ++unk) {
int upos = unk; double ratio;
while (equal(0, equ[upos][unk])) ++upos;
for (int targ = 0; targ < n; ++targ)
if (targ != upos && !equal(0, equ[targ][unk])) {
ratio = equ[targ][unk] / equ[upos][unk];
for (int var = 0; var < 52; ++var)
equ[targ][var] -= equ[upos][var] * ratio;
}
if (upos != unk) std::swap_ranges(equ[upos], equ[upos] + 52, equ[unk]);
ratio = 1 / equ[unk][unk];
if (!equal(1, ratio))
for (int var = 0; var < 52; ++var)
equ[unk][var] *= ratio;
}
}
char a[26], b[26], t[26];
int main() {
memset(equ, 0, sizeof(equ));
int n; scanf("%d", &n);
scanf("%s%s%s", a, b, t);
for (int i = 0; i < n; ++i) {
++equ[i][a[i] - 'A'];
++equ[i][b[i] - 'A'];
--equ[i][t[i] - 'A'];
++equ[i][i + 26];
}
dogauss(n);
int stup = 1 << (n - 1);
for (int viol = 0; viol < stup; ++viol) {
#define clear(x) memset(x, 0, sizeof(x))
clear(var); clear(arg); clear(filt);
bool prev = false;
for (int i = 0; i < n; ++i) {
arg[i] -= prev;
arg[i] += (prev = (viol >> i) & 1) * n;
}
for (int i = 0; i < n; ++i) {
double vi = 0;
for (int j = 0; j < n; ++j)
vi += equ[i][j + 26] * arg[n - j - 1];
if (!equal(0, vi - (var[i] = round(vi)))) goto fail;
if (var[i] >= n || var[i] < 0 || filt[var[i]]) goto fail;
filt[var[i]] = true;
}
for (int i = 0; i < n; ++i) printf("%d ", var[i]);
putchar('\n');
break;
fail:
continue;
}
}