动态规划-最大连续子序列

最大连续子序列

浙江大学/中国矿业大学机试题

目录

最大连续子序列

题目描述

输入输出格式

输入描述:

输出描述:

输入输出样例

输入样例#:

输出样例#:

解题方法

思路

代码

---

题目描述

Time Limit: 1000 ms
Memory Limit: 256 mb

给定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个整数，中间用空格分隔。当K为0时，输入结束，该用例不被处理。

输出描述:

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

输入输出样例

输入样例#:

6
-2 11 -4 13 -5 -2
10
-10 1 2 3 4 -5 -23 3 7 -21
6
5 -8 3 2 5 0
1
10
3
-1 -5 -2
3
-1 0 -2
0

输出样例#:

20 11 13
10 1 4
10 3 5
10 10 10
0 -1 -2
0 0 0

解题方法

思路

dp[i]代表以v[i]为结尾的最大连续子序列的和，用一个maxSum来记录最大值

start和end代表最大子连续序列的第一个和最后一个

若循环过程中dp[i]>maxum成立则当前的v[i]一定是最大连续子序列中的最后一个元素，第一个元素也更新为记录的startIndex对应的元素

代码

#include <bits/stdc++.h>
using namespace std;
int main(){
	int n;
	while(cin>>n){
		if(n==0) break;
		vector<int> v(n);
		int start,end,maxSum,startIndex;
		for(int i=0;i<n;i++){
			cin>>v[i];
		}
		vector<int> dp(n);
		dp[0] = v[0];
		start = end = maxSum = dp[0];
		startIndex =0;
		for(int i=1;i<v.size();i++){
			dp[i] = max(dp[i-1]+v[i],v[i]);

			if(dp[i-1]<0) startIndex = i;
			
			if(dp[i]>maxSum){
				maxSum = dp[i];
				end = v[i]; 
				start = v[startIndex];
				
			}
		}
		if(maxSum<0) {//题目的特别要求
			maxSum=0;
			start = v[0];
			end = v[n-1];
		}
		cout<<maxSum<<" "<<start<<" "<<end<<endl;
	}
	
	return 0;
}