牛客-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）。

輸入描述:

第一行一個整數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的最大值。然后枚舉中間點得到最大值。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
#include<vector>
#include<functional>
using namespace std;
#define LL long long 
#define pii pair<long long ,int>
#define mp make_pair
#define inf 0x3f3f3f3f
#define linf 0xffffffffff
LL a[200020],pre[200020],f1[200020],f2[200020];
int main()
{
    //cout<<linf<<endl;
    int n,m,i,j,k,t;
    cin>>t;
    while(t--){
        cin>>n>>k;
        for(i=1;i<=n;++i) scanf("%lld",a+i),pre[i]=pre[i-1]+a[i],f1[i]=f2[i]=-linf;
    //    memset(f1,-inf,sizeof(f1));
    //    memset(f2,-inf,sizeof(f2));
        for(i=k;i<=n;++i){
            f1[i]=max(f1[i-1],pre[i]-pre[i-k]);
        }
        for(i=n-k+1;i>=1;--i){
            f2[i]=max(f2[i+1],pre[i+k-1]-pre[i-1]);
        }
        LL ans=-linf;
        for(i=k;i<n-k+1;++i){
            ans=max(ans,f1[i]+f2[i+1]);
        }
        cout<<ans<<endl;
    }
    return 0;
}