前缀和-经典题 数学考试

原题目戳这

题目描述

今天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能获得的最大分数

输入
2
6 3
1 1 1 1 1 1
8 2
-1 0 2 -1 -1 2 3 -1
输出
6
7

我的思路：

要考察一段区间的和，很容易想到前缀和：sum[i]=i之前所有数的和。依题意：一段长度为k的区间的所有数之和可表示为：sum[i]-sum[i-k]。有两个区间（姑且分别称之为左区间和右区间），由于它们不能重叠，所以左区间必然在循环变量i的左边，右区间必然在循环变量i的右边。仔细考虑左右边界点，能够想出：循环变量i是从k到n-k的，即从左数、从右数的k个数不考虑。因为我们是要考察长度为k的一段区间的和，长度小于k的显然不合题意。然后找出最大的左区间，依次在i的右边找到一个右区间，使得两区间之和最大即可。

代码：

#include"bits/stdc++.h"
using namespace std;
#define ll long long
const ll inf=0x3f3f3f3f3f3f3f3f;
const int MAX=200005;
int n,k,t,p=0;
ll sum[MAX];
ll anses[MAX];
int main() {
    cin>>t;
    while(t--){
        cin>>n>>k;
        for(int i=1;i<=n;++i) {
            cin>>sum[i];
            sum[i]+=sum[i-1];
        }
        ll ans=-inf,zuo=-inf,you=-inf;
        for(int i=k;i<=n-k;++i){
        	/*
            zuo=max(zuo,sum[i]-sum[i-k]);
            you=max(you,sum[i+k]-sum[i]);
            ans=max(ans,zuo+you);
            */
            //上述代码为什么不行呢？请看A：这种情况下zuo，you最终的值都是88，显然不对
            //假设有：1 2 3 5 6 88 9 10 15---A(模拟sum[i]-sum[i-k]区间之和数组)
            //下面为什么行呢？因为它保证了you一定是在zuo的右边的；
            //当zuo取到最大值88时，i必然在88的右边了
            zuo=max(zuo,sum[i]-sum[i-k]);
            ans=max(ans,zuo+sum[i+k]-sum[i]);

        }
        anses[p++]=ans;
    }
    for(int i=0;i<p;++i){
        cout<<anses[i]<<endl;
    }
    return 0 ;
}