数学考试(前缀和+dp)

数学考试(前缀和+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)
输出描述:
输出一个整数,qwb能获得的最大分数
示例1
输入
复制
2
6 3
1 1 1 1 1 1
8 2
-1 0 2 -1 -1 2 3 -1
输出
复制
6
7

问题分析:

区间长度k固定,求某个区间的和考虑用前缀和。presum[i]=a[i]+presum[i-1]。枚举左边区间和的最大值,然后枚举右边的区间和,当左边的区间和比原来的区间和大的时候则当前区间和接管之间左边的区间和去继续枚举,考虑到当前左区间和最大值+上右区间不一定比之前的结果好,所以应当与之前结果作比较保留较大的结果。
ma=max(ma,presum[i]-presum[i-k])//维护左区间最大值
ans=max(ans,presum[i+k]-presum[i]+ma)//如果ma被接管,则判断新的ma+第一次枚举的右区间值是否比之前的结果好,如果没被接管则继续用原ma枚举右区间。

代码实现:

#include <iostream>
#define ll long long
using namespace std;
int main(int argc, char *argv[]) {
	ll presum[200005]={0};
	int T;
	cin>>T;
	while(T--)
	{
		ll n,k;
		ll i;
		ll ma=-1e18,ans=-1e18;
		cin>>n>>k;
		for(i=1;i<=n;i++)
		{
			cin>>presum[i];
			presum[i]+=presum[i-1];
		}
		for(i=k;i+k<=n;i++)
		{
			ma=max(ma,presum[i]-presum[i-k]);
			ans=max(ans,presum[i+k]-presum[i]+ma);
		}
		cout<<ans<<endl;
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值