数学考试~前缀和(尺取)

链接: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

沉着冷静 认真分析 这题的复杂度用O(n)就可以过了,处理一下就好
注意初始化ans的时候要足够小,一开始一直WA这里

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn = 1e5 + 10;
 4 long long sum[2 * maxn], a[2 * maxn], pre[2 * maxn], premax[2 * maxn];
 5 int main() {
 6     int n, k, t;
 7     scanf("%d", &t);
 8     while(t--) {
 9         scanf("%d%d", &n, &k);
10         sum[0] = 0;
11         for (int i = 1 ; i <= n ; i++) {
12             scanf("%lld", &a[i]);
13             sum[i] = sum[i - 1] + a[i];
14         }
15         premax[0] = -999999999999;
16         for (int i = 1 ; i <= n - k + 1 ; i++ ) {
17             pre[i] = sum[k + i - 1] - sum[i - 1];
18             premax[i] = max(pre[i], premax[i - 1]);
19         }
20         long long ans = -999999999999999;
21         for (int i = n - k + 1 ; i >= k  ; i--  ) {
22             ans = max(ans, pre[i] + premax[i - k]);
23         }
24         printf("%lld
", ans);
25     }
26     return 0;
27 }
原文地址:https://www.cnblogs.com/qldabiaoge/p/8857673.html