牛客1.4数学考试
今天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
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int t;
ll a[N];
ll sum[N];
int main()
{
cin>>t;
while(t--)
{
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sum[i]=sum[i-1]+a[i];
}
if(n==k*2) cout<<sum[n]<<endl;
else
{
ll x=-N,y=-N;
//一个循环 ,便利第一个区间的范围取最大,
//同时遍历第二个区间的范围,结合第一个区间的最大值,取答案最大值
for(int i=k;i<=n-k;i++)
{
x=max(x,sum[i]-sum[i-k]) ;//i为第一个区间的右端点
y=max(y,x+sum[i+k]-sum[i]);//i+1为第二个区间的左端点
}
cout<<y<<endl;
}
}
}