codeforces 332b Maximum Absurdity dp

题意：

有n个数，找两个段（每个段长度为k）不相交，求出这两个段的和的最大值。

输出两个段的起始下标。

思路：

dp[i][j]：走到i，已经选了j个段的最大值。

转移：dp[i][j] = dp[i-k][j-1] + sum[i]-sum[i-k];

输出方案的话，根据dp的值dfs回去（不dfs也是可以的），因为方案已经记录在dp的转移中。

code：

#include <cmath>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <cstdio>
using namespace std;

typedef long long LL;
const int MAXN = 2*1e5+5;
const int INF = 0x3f3f3f3f;

int n, k;
int r1 = INF, r2 = INF;
LL dp[MAXN][3], sum[MAXN];
void dfs(int idx, int ti) {
    if(ti == 0 || idx < k) return ;
    if(dp[idx][ti] == dp[idx-1][ti])
        dfs(idx-1, ti);
    else if(dp[idx][ti] == dp[idx-k][ti-1] + sum[idx]-sum[idx-k]) {
        dfs(idx-k, ti-1);
        if(ti == 2) r2 = min(r2, idx-k+1);
        else r1 = min(r1, idx-k+1);
    }
}
        
int main() {
    scanf("%d%d", &n, &k);
    int tmp;
    for(int i = 1;i <= n; i++) {
        scanf("%d", &tmp);
        sum[i] = sum[i-1]+tmp;
    }
    //
    memset(dp, 0, sizeof(dp));
    for(int i = k;i <= n; i++) {
        for(int j = 1;j <= 2; j++) {
            dp[i][j] = max(dp[i][j], dp[i-1][j]);
            if(j == 1 || dp[i-k][j-1] != 0)
                dp[i][j] = max(dp[i][j], dp[i-k][j-1]+sum[i]-sum[i-k]);
        }
    }
    //path
    dfs(n, 2);
    printf("%d %d\n", r1, r2);
    return 0;
}