Codeforces Round #750 (Div. 2) E. Pchelyonok and Segments

Codeforces Round #750 (Div. 2)的其他题解点我

E. Pchelyonok and Segments

---

题目大意：
在一个数组中找出不相交的k个区间，第一个区间长度为k，使区间长度递减的同时，区间和递增，问最大的k是多少（可以从任意点开始为第一个区间）

思路：
利用前缀和来O（1）求区间和，然后很容易知道k绝对不会超过500（等差数列前n项和）
显然是个区间DP

状态dp[i][k] 为 i - n 取k个区间的区间最大值，我们反过来求递减的

状态转移：

if(dp[i+j][j-1]!=0 && sum[i] - sum[i+j] < dp[i+j][j-1]){
	dp[i][j] = max(dp[i][j], sum[i] - sum[i+j]);
	ans = max(ans, j);
	//ans是记录答案的
}

初始化：

dp[n][1] = a[n];
dp[i][1] = max(dp[i + 1][1], a[i]);

AC代码：

#include <bits/stdc++.h>
#define PII pair<int,int>
#define ll long long

using namespace std;

const double eps = 1e-8;
const int maxn = 1e5 + 10;
const int mod = 1e9 + 7;
const int INF = 1<<30;
inline void swap(int &x, int &y){x^=y^=x^=y;}
inline int gcd(int a,int b) {return !b ? a : gcd(b,a%b);}

ll dp[maxn][505];
ll sum[maxn], a[maxn];
int main(){	
    int T;
    scanf("%d", &T);
    while(T--){
        //memset(dp, 0, sizeof(dp));
        int n;
        scanf("%d", &n);
        int ans = 1;
        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];
        for(int i = n - 1; i >= 1; --i){
            dp[i][1] = max(dp[i + 1][1], a[i]);
            for(int j = 2; j <= 500 && i + j <= n; ++j){
                dp[i][j] = dp[i + 1][j];
                if(dp[i+j][j-1]!=0 && sum[i] - sum[i+j] < dp[i+j][j-1]){
                    dp[i][j] = max(dp[i][j], sum[i] - sum[i+j]);
                    ans = max(ans, j);
                } 
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}