CCF CSP 201809-4 再卖菜

bfs + 记忆化搜索(剪枝）

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