bfs + 记忆化搜索(剪枝)
#include <bits/stdc++.h>
using namespace std;
const int N = 310;
int a[N],b[N],f[N][N][N],n;
bool dfs(int u)
{
if(u == n + 1)
{
if((a[n - 1] + a[n]) / 2 == b[n]) return true;
return false;
}
for(int i = 0; i < 3; ++i)
{
a[u] = 3 * b[u - 1] - a[u - 1] - a[u - 2] + i;
if(a[u] > 0 && !f[u][a[u]][a[u - 1]] && dfs(u + 1)) return true;
f[u][a[u]][a[u - 1]] = 1;
}
return false;
}
int main()
{
cin >> n;
for(int i = 1; i <= n; ++i) cin >> b[i];
for(int i = 1; i <= b[1] * 2; ++i)
{
a[1] = i;
for(int j = 0; j < 2; ++j)
{
a[2] = b[1] * 2 - a[1] + j;
if(a[2] > 0 && dfs(3))
{
for(int i = 1; i <= n; ++i) cout << a[i] << ' ';
return 0;
}
}
}
}