最大子段和,可以O(n)实现。跟HDOJ1003几乎一样,但细节上有些不同。
一定要注意的是,序列全部为负才输出头尾的值,如果出现了-3 -2 -1 0这样的序列,是要输出0 0 0的。
#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e4+10;
int k, arr[maxn], max_sum = 0, first = INF, last = INF;
void read()
{
cin >> k;
for(int i = 0; i < k; ++i)
cin >> arr[i];
}
void solve()
{
int cur_sum = 0, cur_first = INF, cur_last = INF;
for(int i = 0; i < k; ++i)
{
cur_sum += arr[i];
cur_last = i;
if(cur_sum > max_sum)
{
max_sum = cur_sum;
first = cur_first;
last = cur_last;
}
else if(cur_sum == arr[i] && arr[i] == 0) //-3 -2 -1 0的情况
{
first = i;
last = i;
}
if(cur_sum < 0)
{
cur_sum = 0;
cur_first = i+1;
}
}
}
void print()
{
cout << max_sum << ' ';
if(first < k) cout << arr[first];
else cout << arr[0];
cout << ' ';
if(last < k) cout << arr[last];
else cout <<arr[k-1];;
}
int main()
{
read();
solve();
print();
return 0;
}