洛谷 P1651 塔

洛谷 P1651 塔

洛谷传送门

题目描述

小明很喜欢摆积木，现在他正在玩的积木是由N个木块组成的，他想用这些木块搭出两座高度相同的塔，一座塔的高度是搭建它的所有木块的高度和，并且一座塔至少要用一个木块。每个木块只能用一次，也可以不用。目前已知每块木块的高度，小明想知道在最终两个塔的高度相同的情况下，他所能搭的塔的最大高度是多少，你能帮助他吗？

输入格式

第一行为一个整数N，表示木块个数。

第二行是N个整数，表示N块木块的高度。

【数据规模】

对于100%的数据，N≤50，每块木块的高度h满足1≤h≤500000，所有木块的高度总和≤500000。

输出格式

仅一个整数，表示能搭建的塔的最大高度，若不能搭建两座相同高度的塔，则输出“-1”。

---

题解：

差值DP模板。

加了个滚动数组。

嗖嗖快。

提示：滚动数组要开大一些，否则容易RE。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=5e5+5;
int n,sum,ans=-1;
int h[maxn];
int dp[3][maxn];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&h[i]);
		sum+=h[i];
	}
	memset(dp,-0x3f,sizeof(dp));
	dp[0][0]=0;
	for(int i=1;i<=n;i++)
		for(int j=0;j<=sum;j++)
		{
			dp[i%2][j]=max(dp[(i-1)%2][j],dp[(i-1)%2][j+h[i]]);
			if(j>=h[i])
				dp[i%2][j]=max(dp[i%2][j],dp[(i-1)%2][j-h[i]]+h[i]);
			else
				dp[i%2][j]=max(dp[i%2][j],dp[(i-1)%2][h[i]-j]+j);
		}
	if(dp[n%2][0]>0)
		printf("%d",dp[n%2][0]);
	else
		puts("-1");
	return 0;
}