题意:
N个整数组成的序列a[1],a[2],a[3],…,a[n],从中选出一个子序列(a[i],a[i+1],…a[j]),使这个子序列的和>0,并且这个和是所有和>0的子序列中最小的。
例如:4,-1,5,-2,-1,2,6,-2。-1,5,-2,-1,序列和为1,是最小的。
Input
第1行:整数序列的长度N(2 <= N <= 50000) 第2 - N+1行:N个整数
Output
输出最小正子段和。
Input示例
8 4 -1 5 -2 -1 2 6 -2
Output示例
1
思路:
保存前缀和sum,对于每一个sum[i],按照题目要求就是要找出比sum[i]小的最接近的一个sum[j],利用set保存之前遍历过的sum。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int MAXN = 50005;
ll a[MAXN], sum[MAXN];
set <ll> :: iterator it;
int main() {
int n;
scanf("%d", &n);
set <ll> st;
for (int i = 1; i <= n; i++) {
scanf("%I64d", &a[i]);
sum[i] = sum[i - 1] + a[i];
}
ll ans = INF;
st.insert(0);
for (int i = 1; i <= n; i++) {
it = st.lower_bound(sum[i]);
// printf("%I64d %I64d\n", *it, sum[i]);
if (it == st.begin()) {
if (a[i] > 0) ans = min(ans, a[i]);
}
else ans = min(ans, sum[i] - *(--it));
st.insert(sum[i]);
}
printf("%I64d\n", ans);
return 0;
}