PTA甲级1007 动态规划

PTA甲级1007 动态规划

题目

大意：给一个序列，求i，j，使得ai+···+aj最大，输出最大和及ai，aj。

思路：一个简单的动态规划 主要分成最大和问题和输出最大子序列首尾元素，多设置一个是s[]存放下标即可，特殊情况为全负数，放在一开始处理，直接输出0和首尾元素。

代码

#include<cstdio>
const int maxn = 10010;
int a[maxn],dp[maxn];//a[i]存放序列，dp[i]存放a[i]结尾的连续序列最大和
int s[maxn]={0};//表示dp从a的哪个元素开始
int main(){
    int n;
    scanf("%d",&n);
    bool flag = false;//查看元素是否全小于0
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
        if(a[i]>=0) flag = true;//只要有一个数大于0，就记为ture
    }
    if(flag == false){//如果都小于0，输出0及首尾元素
        printf("0 %d %d",a[0],a[n-1]);
        return 0;
    }
    dp[0] = a[0];
    for(int i =1;i<n;i++){
        if(dp[i-1]+a[i]>a[i]){
            dp[i] = dp[i-1]+a[i];
            s[i]=s[i-1];
        }else{
            dp[i]=a[i];
            s[i]=i;
        }
    }
    int k=0;
    for(int i = 1;i<n;i++){
        if(dp[i]>dp[k]){
            k=i;
        }
    }
    printf("%d %d %d",dp[k],a[s[k]],a[k]);
    return 0;
}