题目大意:计算给定序列的最大子序列和,并输出这个最大子序列的首尾元素,如果有多个最大子序列存在,输出第一个。注意,如果序列中所有的元素都是负数,那么输出0,以及整个序列的首尾元素。
使用在线处理方法计算最大子序列和,即当前和子序列和为负数时,直接置子序列和为0,并从下一个元素开始。同时还要记录子序列的首尾元素,尾元素只需要在更新最大和的时候记录即可,而当前子序列的首元素则需要在每一次序列和被重置为0的时候记录(包括开始时序列和被初始化为0),然后在更新最大和的时候更新最大子序列的首元素。坑点是所有元素都为负的时候需要特判输出。
AC代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>
using namespace std;
int main()
{
int K;
cin >> K;
int sum = 0, maxSum = 0, fistNum = 0, lastNum = 0, start = 0, firstInput = 0, lastInput = 0;
bool allNegtive = true;
for (int i = 0; i < K; ++i)
{
int num;
scanf("%d", &num);
if(num >= 0) allNegtive = false;
if(sum == 0) start = num;
sum += num;
if(sum < 0) sum = 0;
else if(sum > maxSum)
{
maxSum = sum;
fistNum = start;
lastNum = num;
}
if(i == 0) firstInput = num;
else if(i == K - 1) lastInput = num;
}
if(!allNegtive) printf("%d %d %d", maxSum, fistNum, lastNum);
else printf("0 %d %d", firstInput, lastInput);
return 0;
}