P1651 塔-【普及+提高】DP数组的平移与二维跳跃

塔 - 洛谷

设dp[i][j]为前i个木块拼成了两座塔，左边减去右边差值为j时，左边塔的最高高度。先忽略正负与空间问题，第i块木块，可以放也可以不放，不放就要承接前i-1块差值为j的状态，放的话，可以放在右边，使得差值为j+a[i]的变成j,也可以放在左边，使得差值j-a[i]的变成j.

但我们发现，存在差值为负的情况，所以我们动用了数组平移。设置500000为差值为0点，-500000为差值负向最大点，500000为差值正向最大点

另外，本次状态只受前一状态影响，连续开辟50*500000的数组一定会mle,故采用类似于一维滚动数组的二维跳跃数组，i=1时，采用i%2(1)这一维，该维度由(i%2)^1(0)转移而来，并且在此基础上进行了优化。i=2时，使用第二维，借助第一位状态转移，i=3时，使用第一维，这时借助第二维转移，无须担心第一位状态未清零，因为我们要借助第二维转移，而第二维由第一维转移而来，只会更优。

关于答案输出，只需找到差值为0时，左边塔高度即可

#include <iostream>
#include <string.h>
#include <algorithm>
#include <cstdio>
#include <stdlib.h>
#include <string>
#include <memory>
#include <queue>
# include<math.h>
# include<iomanip>
# include<map>
# define mod 1073741824
# include<math.h>
using namespace std;

typedef long long int ll;
typedef double dl;

int dp[2][500000*3+10];
int a[100];


int main()
{

    int n;
    cin>>n;

    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }

    int sum=0;

    int ans=0;

    fill(dp[0],dp[0]+2*500000*2+10,-9999999);

    dp[0][500000]=0;
    for(int i=1;i<=n;i++)
    {

        sum+=a[i];

        for(int j=500000-sum;j<=500000+sum;j++)
        {


        dp[i%2][j]=dp[(i%2)^1][j];

        dp[i%2][j]=max(dp[i%2][j],dp[(i%2)^1][j-a[i]]+a[i]);

        dp[i%2][j]=max(dp[i%2][j],dp[(i%2)^1][j+a[i]]);

       if(j==500000)
        ans=max(ans,dp[i%2][j]);

        }
    }

    if(ans>0)
        cout<<ans;
    else
        cout<<-1;



    return 0;
}