网易2017春招 堆砖块(dp+滚动数组)

时间限制：1秒
空间限制：32768K

小易有n块砖块，每一块砖块有一个高度。小易希望利用这些砖块堆砌两座相同高度的塔。为了让问题简单，砖块堆砌就是简单的高度相加，某一块砖只能使用在一座塔中一次。小易现在让能够堆砌出来的两座塔的高度尽量高，小易能否完成呢。

输入描述:

输入包括两行： 第一行为整数n(1 ≤ n ≤ 50)，即一共有n块砖块 第二行为n个整数，表示每一块砖块的高度height[i] (1 ≤
height[i] ≤ 500000)

输出描述:

如果小易能堆砌出两座高度相同的塔，输出最高能拼凑的高度，如果不能则输出-1. 保证答案不大于500000。

输入例子1:

3 2 3 5

输出例子1:

5

dp[i][j]

i表示前i块砖，j表示A堆与B堆的差值，dp[i][j]表示前i块砖，差值为j时，其中一堆的大小(这里我设为A堆)

设 sum为所有砖的总和

那么一共有三种情况：
case1 i砖放A堆: dp[i][j]=dp[i-1][j-w[i]]+w[i]
case2 i砖放B堆: dp[i][j]=dp[i-1][j+w[i]]
case3 不放i砖: dp[i][j]=dp[i-1][j]

刷新最大值即可
这里还有个坑点容易想不到：由于j为两堆砖的差值，但差值可能为负数，因此要把sum变大为原来的2倍，前1~sum-1表示负数，sum表示0，后sum+1~sum*2表示正数

事情到此就结束了了吗？没有！由于sum范围5e5,如果前面n开50的话肯定要爆内存，观察dp式子，发现，每个i都由前面的i-1得来，那么，只需要用滚动数组刷新即可

#include<bits/stdc++.h>
using namespace std;
const int N=500000;
int w[55],dp[3][2*N+500];
int main()
{
    int n,sum=0;
    cin>>n;
    memset(dp,-1,sizeof(dp));
    for(int i=1; i<=n; i++)
    {
        cin>>w[i];
        sum+=w[i];
    }
    dp[0][sum]=0;
    int k=0;
    for(int i=1; i<=n; i++)
    {
        for(int j=0; j<=2*sum; j++)
        {
            dp[k^1][j]=dp[k][j];
            if(j-w[i]>=0&&dp[k][j-w[i]]!=-1)
                dp[k^1][j]=max(dp[k^1][j],dp[k][j-w[i]]+w[i]);
            if(j+w[i]<=2*sum&&dp[k][j+w[i]]!=-1)
                dp[k^1][j]=max(dp[k^1][j],dp[k][j+w[i]]);
        }
        k^=1;
    }
    if(dp[n&1][sum]==0)puts("-1");
    else printf("%d\n",dp[n&1][sum]);
    return 0;
}