洛谷 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;
}