题意
给定
n
n
n 和
n
n
n个整数
a
i
(
1
≤
i
≤
n
,
−
1
0
4
≤
a
i
≤
1
0
4
)
a_i(1 \leq i \leq n,-10^4 \leq a_i \leq 10^4)
ai(1≤i≤n,−104≤ai≤104),请选出其中连续且非空的一段使得这段和最大,并它的和。
思路
前言:如果不会前缀和的请移步我的这篇前缀和、差分简析,谢谢!
考虑题目要求 ∑ i = x y a i \sum_{i = x}^{y}{a_i} ∑i=xyai 的最大值,并且要用 O ( n ) O(n) O(n) 的时间复杂度,因此优先构造前缀和数组 b i ← ∑ j = 1 i a j b_i \gets \sum_{j = 1}^{i}{a_j} bi←∑j=1iaj。
题目要求的 ∑ i = x y a i = b y − b x \sum_{i = x}^{y}{a_i} = b_y - b_x ∑i=xyai=by−bx。因为 b y b_y by 可以循环枚举,考虑在 O ( 1 ) O(1) O(1) 时间复杂度内找到 x x x 让 b x b_x bx 最小。令 m i n n minn minn 为当前最小的 b x b_x bx, a n s ans ans 为最后的答案:
- a n s ← max ( a n s , b i − m i n n ) ans \gets \max(ans,b_i - minn) ans←max(ans,bi−minn)
- m i n n ← m i n ( m i n n , b i ) minn \gets min(minn,b_i) minn←min(minn,bi)。
因为 m i n n minn minn 为当前最小的 b x b_x bx,所以 m i n n minn minn 初始值为 0 0 0。
代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,a[200005],b[200005],minn = 0,ans = -1e18;
signed main() {
scanf("%lld",&n);
for(int i = 1;i <= n;i++) scanf("%lld",&a[i]),b[i] = b[i - 1] + a[i];
for(int i = 1;i <= n;i++) {
ans = max(ans,b[i] - minn);
minn = min(minn,b[i]);
}
printf("%lld\n",ans);
return 0;
}