链接:https://ac.nowcoder.com/acm/problem/15553
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
今天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
WA因:求出前缀和后打算暴力二重循环求两个区间最大,tle
解法:一重循环,在过程中维护左侧最大前缀和与当前数相加,保存最大
ll max0=-1e18,max1=-1e18;
for(int i=2*k;i<=n;++i)
{
max1=max(max1,b[i-k]);//左侧最大
max0=max(max0,max1+b[i]);//相加最大
}
AC码:
#include<bits/stdc++.h>
#define nMax 200010
typedef long long ll;
using namespace std;
ll a[nMax];
ll b[nMax];
int main()
{
int t;
cin>>t;
while(t--)
{
int n,k;
cin>>n>>k;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
for(int i=1;i<=n;++i)
{
scanf("%lld",&a[i]);
a[i]+=a[i-1];
}
for(int i=k;i<=n;++i)
{
b[i]=a[i]-a[i-k];
}
ll max0=-1e18,max1=-1e18;
for(int i=2*k;i<=n;++i)
{
max1=max(max1,b[i-k]);
max0=max(max0,max1+b[i]);
}
cout<<max0<<endl;
}
}