51Nod 1052 - 最大M子段和(DP)
阿新 • • 發佈:2018-11-09
題目連結 http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1052
【題目描述】
N個整陣列成的序列a[1],a[2],a[3],…,a[n],將這N個數劃分為互不相交的M個子段,並且這M個子段的和是最大的。如果M >= N個數中正數的個數,那麼輸出所有正數的和。
例如:-2 11 -4 13 -5 6 -2,分為2段,11 -4 13一段,6一段,和為26。
Input
第1行:2個數N和M,中間用空格分隔。N為整數的個數,M為劃分為多少段。(2 <= N , M <= 5000)
第2 - N+1行:N個整數 (-10^9 <= a[i] <= 10^9)
Output
輸出這個最大和
Input示例
7 2
-2
11
-4
13
-5
6
-2
Output示例
26
【思路】
設
表示前
個元素劃分成
段,且包含第
個元素對應的答案,那麼根據第
的元素所處的段,有狀態轉移
邊界是
(這道題的答案都是正數,別的題邊界可能不太一樣)
用滾動陣列比較好寫,只記錄當前行和上一行的答案就可以,然後在計算過程中,用一個變數 記錄上一行 到 的最大值並不斷更新
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf=1e18;
const int maxn=5005;
int n,m;
ll a[maxn];
ll dp[2][maxn];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
for(int i=1;i<=m;++i){
int cur=i%2;
int pre=1-cur;
ll maxpre=dp[pre][i-1];
for(int j=i;j<=n;++j){
dp[cur][j]=max(dp[cur][j-1],maxpre)+a[j];
maxpre=max(maxpre,dp[pre][j]);
}
}
ll ans=-inf;
for(int j=m;j<=n;++j){
ans=max(ans,dp[m%2][j]);
}
printf("%lld\n",ans);
return 0;
}