题目描述
给定一个包含 K 个整数的序列 {N1,N2,…,NK}。
连续子序列定义为 {Ni,Ni+1,…,Nj},其中 1≤i≤j≤K。
最大子序列是指序列内各元素之和最大的连续子序列。
例如,给定序列 {−2,11,−4,13,−5,−2},它的最大子序列为 {11,−4,13},其各元素之和为 20。
现在你需要求出最大子序列的各元素之和,并且输出最大子序列的第一个元素和最后一个元素的值。
输入格式
第一行包含一个整数 K。
第二行包含 K 个整数。
输出格式
输出一行三个整数,分别表示最大子序列的各元素之和以及最大子序列的第一个元素和最后一个元素的值。
设最大子序列为 {Ni,Ni+1,…,Nj},如果答案不唯一,则选择 i 更小的解,如果仍不唯一,则选择 j 更小的解。
注意,我们规定,如果所有 K 个数字均为负数,则其最大和定义为 0,并且应该输出整个序列的第一个数字和最后一个数字。
数据范围
1≤K≤10000,
序列内元素的绝对值不超过 105。
样例
输入样例:
10
-10 1 2 3 4 -5 -23 3 7 -21
输出样例:
10 1 4
算法
(动态规划) $O(n)$
#include <iostream>
using namespace std;
const int N = 10000 + 10;
int a[N];
struct {
int l, r;
int sum;
} dp[N];
// dp[i]:
// sum 以 a[i] 结尾的连续子序列最大和;
// l 连续子序列的开始下标,r= 连续子序列的结束下标 (前闭后闭区间)
//
// dp[i] = max(dp[i - 1] + a[i], a[i])
// ans = max(dp[0], ..., dp[n]);
int main() {
int k;
cin >> k;
for (int i = 0; i < k; i ++) {
cin >> a[i];
}
// dp[i] = max(dp[i - 1] + a[i], a[i])
dp[0].l = 0, dp[0].r = 0, dp[0].sum = a[0];
for (int i = 1; i < k; i ++) {
int l = dp[i - 1].l, r = dp[i - 1].r, sum = dp[i - 1].sum;
// 注意 sum + a[i] == a[i] 时的更新情况
if (sum + a[i] >= a[i]) {
dp[i].sum = sum + a[i];
dp[i].l = l;
dp[i].r = i;
} else {
dp[i].sum = a[i];
dp[i].l = i;
dp[i].r = i;
}
}
// ans = max(dp[0], ..., dp[n]);
int ans = -1e9, l = 0, r = 0;
for (int i = 0; i < k; i ++) {
if (ans < dp[i].sum) {
ans = dp[i].sum;
l = dp[i].l;
r = dp[i].r;
}
}
if (ans < 0) {
cout << "0 " << a[0] << " " << a[k - 1];
} else {
cout << ans << " " << a[l] << " " << a[r];
}
}