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;
}