1. 程式人生 > >51Nod 1052 - 最大M子段和(DP)

51Nod 1052 - 最大M子段和(DP)

題目連結 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

【思路】
d p [ i ] [ j ]

dp[i][j] 表示前 j j 個元素劃分成 i i 段,且包含第 j
j
個元素對應的答案,那麼根據第 j j 的元素所處的段,有狀態轉移 d p [ i ] [ j ] = { d p [ i ] [ j 1 ] + a [ j ]       ( a [ j ] i ) m a x { d p [ i 1 ] [ k ] + a [ j ] i 1 &lt; = k &lt; j }       ( a [ j ] ) dp[i][j]=\begin{cases} dp[i][j-1]+a[j] \ \ \ \ \ (a[j]放到原來第i段的最後) \\ max\{dp[i-1][k]+a[j] | i-1&lt;=k&lt;j\} \ \ \ \ \ (a[j]自己單獨為一段) \end{cases} 邊界是 d p [ 0 ] [ j ] = 0 dp[0][j]=0 (這道題的答案都是正數,別的題邊界可能不太一樣)

用滾動陣列比較好寫,只記錄當前行和上一行的答案就可以,然後在計算過程中,用一個變數 m a x p r e maxpre 記錄上一行 d p [ p r e ] [ i 1 ] dp[pre][i-1] d p [ p r e ] [ j ] dp[pre][j] 的最大值並不斷更新

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