数学考试 (区间DP)

数学考试

题目描述：
今天qwb要参加一个数学考试，这套试卷一共有n道题，每道题qwb能获得的分数为ai，qwb并不打算把这些题全做完，他想选总共2k 道题来做，并且期望他能获得的分数尽可能的大，他准备选2个不连续的长度为k的区间 , 即[L,L+1,L+2,…,L+k-1]，[R,R+1,R+2,…,R+k-1]（R >= L+k）。
输入描述：
第一行一个整数T（T<=10）,代表有T组数据
接下来一行两个整数n,k,(1<=n<=200,000),(1<=k,2k <= n)
接下来一行n个整数a1,a2,…,an，（-100,000<=ai<=100,000）

思路：很明显的DP，复杂度O(n)，此题解法很多，核心是遍历时维护第一或者第二个区间的最值（二选一），然后去枚举另一个区间，再取最值即可。

Code：

#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long LL;
const int N = 2e5+7;

LL dp[N];
LL sum[N];

int main()
{
	int T;
	cin>>T;
	while(T--){
		int n,k,t;
		cin>>n>>k;
		memset(dp,-0x3f,sizeof dp);
		memset(sum,0,sizeof sum);
		for(int i=1;i<=n;i++){
			scanf("%d",&t);
			sum[i] = sum[i-1] + t;		//前缀和
		}
		for(int i=n-k;i>=k;i--){
			dp[i] = max(dp[i+1],sum[i+k]-sum[i]);	// 维护 i 右边区间长度为 k的最大值
		}										//或者在枚举的同时，维护左边区间的最值；
		LL ans = -1e18;
		for(int i=k;i+k<=n;i++){
			ans = max(ans, sum[i]-sum[i-k] + dp[i]);	//  取两个区间和最大的
		}
		cout<<ans<<endl;
	}
	return 0;
}