牛客——数学考试

链接：数学考试 (nowcoder.com)
来源：牛客网

题目描述

今天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,(2<=n<=200,000),(1<=k,2k <= n)
接下来一行n个整数a1,a2,...,an，（-100,000<=ai<=100,000）

输出描述:

输出一个整数，qwb能获得的最大分数

这题求最大两个不连续区间的和，可以用前缀和来写。 

我的思路是先求出两个区间挨在一起的情况，求出最大的，然后在求出两个区间不挨在一起的情况两者的和，然后将两者进行比较，取最大的。

但看了大佬的代码，好像不用这么麻烦 Ort.... 

本弱鸡的代码：

#include<iostream>
#include<string>
using namespace std;
const int N = 2e5+5;
typedef long long LL;
LL nums[N], sum[N] {0};
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    while(t --)
    {
        int n, k;
        cin >> n >> k;
        for(int i=1; i<=n; i++) cin >> nums[i], sum[i] = sum[i-1] + nums[i]; //求前缀和 
        LL comb, lsum, rsum, indexl = 0, indexr = 0; //分别表挨在一起情况两区间最大和，不挨在一起情况左边区间最大和和右区间最大和
													 //indexl 和 indexr记录不挨在一起情况时，左边区间的左边界和右边界，防止求解右边区间时，两区间重合。 
        comb = lsum = rsum = -1e18;
        for(int i=1; i+2*k-1<=n; i++) //求挨在一起情况时，两区间最大和 
            comb = max(comb, sum[i+2*k-1]-sum[i-1]);

        for(int i=1; i+k-1<=n; i++) //不挨在一起时，求左区间 
        {
            if(sum[i+k-1]-sum[i-1]>lsum)
            {
                lsum = max(lsum, sum[i+k-1]-sum[i-1]);
                indexl = i, indexr = i+k-1; //记录区间边界 
            }
        }
        
        for(int i=1; i+k-1<=n; i++) //不挨在一起时，求右区间 
        {
            if(indexl<=i && i<=indexr || i+k-1>=indexl && i+k-1<=indexr) continue; //如果和左区间重合，则跳过 
            rsum = max(rsum, sum[i+k-1]-sum[i-1]);
        }
        cout << max(lsum+rsum, comb) << endl; //求两种情况最大的那个值 
    }
    return 0;
}

 按模仿大佬写的代码：

#include<iostream>
using namespace std;
typedef long long LL;
const int N = 2e5 + 5;
LL sum[N], nums[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    while(t --)
    {
        int n, k;
        cin >> n >> k;
        for(int i=1; i<=n; i++) cin >> nums[i], sum[i] = sum[i-1] + nums[i];
        LL res, maxv;
        res = maxv = -1e18;
        for(int i=k; i+k<=n; i++) //这里大佬设计得太巧妙了！ 
        {
            maxv = max(maxv, sum[i]-sum[i-k]);//求第一个区间最大和
            res = max(res, maxv + sum[i+k]-sum[i]);//求第二个区间和第一个区间的最大和。
												//因为第二个区间左边界大于第一区间的右边界，所以不用担心重合 
        }
        cout << res << endl;
    }
    return 0;
}