dfs的时候注意记忆化搜索,不然会T掉只有80分。
记忆化搜索的时候保存一个数组
f
i
,
j
,
k
f_{i,j,k}
fi,j,k 保存填第
i
i
i 天时,前两天所填的数字有没有被搜索过
#include <cstdio>
#include <cstdlib>
const int MAXN = 300 + 5;
int n, cnt;
int a[MAXN], b[MAXN];
bool f[MAXN][MAXN][MAXN];
bool check() {
for (int i = 1; i <= n; i++) {
int tmp;
if (i == 1) tmp = (a[i] + a[i + 1]) / 2;
else if (i == n) tmp = (a[i] + a[i - 1]) / 2;
else tmp = (a[i] + a[i + 1] + a[i - 1]) / 3;
if (tmp != b[i]) return false;
}
return true;
}
void dfs(int cur) {
if (cur > n) {
if (check()) {
for (int i = 1; i <= n; i++)
printf("%d ", a[i]);
exit(0);
}
else return;
}
if (cur > 2) {
if (f[cur][a[cur - 2]][a[cur - 1]]) return;
f[cur][a[cur - 2]][a[cur - 1]] = true;
}
if (cur == 2) {
for (int i = 2 * b[1] - a[1]; (a[1] + i) / 2 == b[1]; i++) {
if (i > 0) {
a[cur] = i;
dfs(cur + 1);
}
}
}
else if (cur == n) {
for (int i = 2 * b[n] - a[n - 1]; (a[n - 1] + i) / 2 == b[n]; i++) {
if (i > 0) {
a[cur] = i;
dfs(cur + 1);
}
}
}
else {
for (int i = 3 * b[cur - 1] - a[cur - 2] - a[cur - 1]; (a[cur - 2] + a[cur - 1] + i) / 3 == b[cur - 1]; i++) {
if (i > 0) {
a[cur] = i;
dfs(cur + 1);
}
}
}
}
int main()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%d", &b[i]);
for (int i = 1; ; i++) {
a[1] = i;
dfs(2);
}
return 0;
}