最大子段和
题目描述
给出一个长度为 nnn 的序列 aaa,选出其中连续且非空的一段使得这段和最大。
输入格式
第一行是一个整数,表示序列的长度 nnn。
第二行有 nnn 个整数,第 iii 个整数表示序列的第 iii 个数字 aia_iai。
输出格式
输出一行一个整数表示答案。
样例 #1
样例输入 #1
7
2 -4 3 -1 2 -4 3
样例输出 #1
4
提示
样例 1 解释
选取 [3,5][3, 5][3,5] 子段 {3,−1,2}\{3, -1, 2\}{3,−1,2},其和为 444。
数据规模与约定
- 对于 40%40\%40% 的数据,保证 n≤2×103n \leq 2 \times 10^3n≤2×103。
- 对于 100%100\%100% 的数据,保证 1≤n≤2×1051 \leq n \leq 2 \times 10^51≤n≤2×105,−104≤ai≤104-10^4 \leq a_i \leq 10^4−104≤ai≤104。
思路: 动态规划,设dp[i]为第i个元素前的最大子段和,则状态转移方程为:
dp[i]=max(a[i], dp[i - 1] + a[i])
代码:
#include <bits/stdc++.h>
using namespace std;
int n, a[200005], ans[200005];
int main(){
cin >> n;
for(int i = 0;i < n; i++){
cin >> a[i];
}
ans[0] = a[0];
for(int i = 1;i < n; i++){
ans[i] = max(a[i], ans[i - 1] + a[i]);
}
sort(ans, ans + n);
cout << ans[n - 1];
return 0;
}