首页 > 技术文章 > CodeUp 问题 A: 最大连续子序列

alg-carlossxu 2021-06-18 14:38 原文

题目描述
给定K个整数的序列{ N1, N2, ..., NK },其任意连续子序列可表示为{ Ni, Ni+1, ..., Nj },其中 1 <= i <= j <= K。最大连续子序列是所有连续子序列中元素和最大的一个,例如给定序列{ -2, 11, -4, 13, -5, -2 },其最大连续子序列为{ 11, -4, 13 },最大和为20。现在增加一个要求,即还需要输出该子序列的第一个和最后一个元素。

输入
测试输入包含若干测试用例,每个测试用例占2行,第1行给出正整数K( K<= 10000 ),第2行给出K个整数,中间用空格分隔,每个数的绝对值不超过100。当K为0时,输入结束,该用例不被处理。

输出
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。

我们可以设置结构体数组dp,数组中存储dp值和此处序列首索引和尾索引,再根据状态转移方程来判断如何更新dp数组

代码如下:

#include <bits/stdc++.h>
using namespace std;
#define N 10005
int k;
struct node{	//利用s和e存储该序列首索引和尾索引 
	int v;
	int s,e;
}dp[N];
int T[N];
int main(){
	while(scanf("%d",&k) != EOF, k){
		bool flag = false; //用flag标记输入数据是否都为负 
		for(int i=0;i<k;i++){
			scanf("%d",&T[i]);
			if(T[i] > 0) flag = true;
		}
		if(flag){
			dp[0].v = T[0];
			dp[0].s = dp[0].e = 0;
			for(int i=1;i<k;i++){
				if(dp[i-1].v + T[i] < T[i]){
					dp[i].s = dp[i].e = i;
					dp[i].v = T[i];
				}else{
					dp[i].s = dp[i-1].s;
					dp[i].e = i;
					dp[i].v = dp[i-1].v + T[i]; 
				}
			}
			int MAX = dp[0].v, u = 0;
			for(int i=1;i<k;i++){
				if(dp[i].v > MAX){
					MAX = dp[i].v;
					u = i; 
				}
			}
			printf("%d %d %d\n",dp[u].v,T[dp[u].s],T[dp[u].e]); 			
		}else{
			printf("0 %d %d\n",T[0],T[k-1]);
		}
	}
	return 0;
}

推荐阅读