2021-10-31

E. Pchelyonok and Segments

题目大意：
在一个长度位n的数组中找出不相交的k个区间，第一个区间长度为k，使区间长度递减并且区间和递增，问最大的k是多少？

思路：
区间DP，设状态dp[i][k]为区间[i, n]中取了k个区间的最前面区间的可能的最大值，详细请看代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
ll a[N], dp[N][600], sum[N];//sum会超上限

void solve()
{
//    memset(dp, 0, sizeof(dp));用memset会超时
    int n, ans = 1;
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++)
    {
        scanf("%lld", &a[i]);
        for(int j = 0; j <= 500; j ++)
            dp[i][j] = 0;
    }
        
    
    for(int i = n; i >= 1; i --)
        sum[i] = sum[i + 1] + a[i];//反着求前缀和，后面比较方便
    
    dp[n][1] = a[n];//区间[n, n]只取同一个区间当然是a[n]
    for(int i = n - 1; i >= 1; i --)
    {
        dp[i][1] = max(dp[i + 1][1], a[i]);//区间[i, n]可以从
        //区间[i + 1, n]转移过来。但是只取一个区间要判当前a[i]取不取
        for(int j = 2; j <= 500 && i + j <= n; j ++)
        {
            dp[i][j] = dp[i + 1][j];//直接转移过来
            if(sum[i] - sum[i + j] < dp[i + j][j - 1])
            {//也可能从取了j-1段的转移过来
                dp[i][j] = max(dp[i][j], sum[i] - sum[i + j]);
                ans = max(ans, j);
            }
        }
    }

    printf("%d\n", ans);
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
        solve();
    return 0;
}