HDU 1231 最大連續子序列(DP)

給定K個整數的序列{ N1, N2, ..., NK },其任意連續子序列可表示爲{ Ni, Ni+1, ..., Nj },其中 1 <= i <= j <= K。最大連續子序列是所有連續子序列中元素和最大的一個, 例如給定序列{ -2, 11, -4, 13, -5, -2 },其最大連續子序列爲{ 11, -4, 13 },最大和爲20。 

要求得到最大和,並輸出該子序列的第一個和最後一個元素。

思路是DP,用f [ i ]表示以第 i 個數字爲結尾的子序列的最大和, 以m[ i ] 表示第 i 個數。那麼 f [ i ] = max( f[i - 1], m[ i ] )。

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;

int dp[11000];
int head[11000];
int input[11000];
int main(){
	int K;
	while(scanf("%d", &K)){
		if(!K)
			break;
		bool allNeg = true;
		dp[0] = -1;
		for(int i = 1; i <= K; i++){
			scanf("%d", &input[i]);
			if(input[i] >= 0)
				allNeg = false;
			if(input[i] + dp[i - 1] >= input[i]){
				dp[i] = dp[i - 1] + input[i];
				head[i] = head[i - 1];
			}else{
				dp[i] = input[i];
				head[i] = i;
			}
		}
		if(allNeg){
			printf("0 %d %d\n", input[1], input[K]);
			continue;
		}
		int MAX = dp[1];
		int h = head[1], t = 1;
		for(int i = 1; i <= K; i++){
			if(dp[i] > MAX){
				MAX = dp[i];
				h = head[i];
				t = i;
			}
		}
		printf("%d %d %d\n", MAX, input[h], input[t]);
	}
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章