牛客 數學考試

鏈接：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;
 }
}