牛客-https://www.nowcoder.com/acm/contest/96/H
链接:https://www.nowcoder.com/acm/contest/96/H
来源:牛客网
题目描述
今天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)。
他想选总共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
dp预处理出来f1,f2数组,f1[i]表示a[1..i]中长度为k的最大值,f2[i]表示a[i..n]中长度为k的最大值。然后枚举中间点得到最大值。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<queue> 5 #include<cmath> 6 #include<algorithm> 7 #include<map> 8 #include<set> 9 #include<vector> 10 #include<functional> 11 using namespace std; 12 #define LL long long 13 #define pii pair<long long ,int> 14 #define mp make_pair 15 #define inf 0x3f3f3f3f 16 #define linf 0xffffffffff 17 LL a[200020],pre[200020],f1[200020],f2[200020]; 18 int main() 19 { 20 //cout<<linf<<endl; 21 int n,m,i,j,k,t; 22 cin>>t; 23 while(t--){ 24 cin>>n>>k; 25 for(i=1;i<=n;++i) scanf("%lld",a+i),pre[i]=pre[i-1]+a[i],f1[i]=f2[i]=-linf; 26 // memset(f1,-inf,sizeof(f1)); 27 // memset(f2,-inf,sizeof(f2)); 28 for(i=k;i<=n;++i){ 29 f1[i]=max(f1[i-1],pre[i]-pre[i-k]); 30 } 31 for(i=n-k+1;i>=1;--i){ 32 f2[i]=max(f2[i+1],pre[i+k-1]-pre[i-1]); 33 } 34 LL ans=-linf; 35 for(i=k;i<n-k+1;++i){ 36 ans=max(ans,f1[i]+f2[i+1]); 37 } 38 cout<<ans<<endl; 39 } 40 return 0; 41 }