1. 程式人生 > 其它 >洛谷 [NOIP2009 普及組] 道路遊戲(dp)

洛谷 [NOIP2009 普及組] 道路遊戲(dp)

傳送門


解題思路

dp[i][j][k]表示從第i秒開始走走j秒從k工廠開始走的收益
maxx[i][j]表示從第i秒開始走j秒的最大收益。
則轉移方程為:
dp[i][j][k]=dp[i+1][j-1][(k+1)%n]+value[k][i];
maxx[i][j]=max(dp[i][j][k]-cost[k]);
其中i這一維可以滾動陣列優化掉。

做完上述操作後,令
f[i]表示前i秒的最大收益,則
f[i]=max(f[j-1]+maxx[j][i-j+1]);

最後f[m]即為答案。

注意初始化f和maxx為大負數。

AC程式碼

#include<cstdio>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=1005;
int n,m,p,a[maxn][maxn],dp[2][maxn][maxn],f[maxn],maxx[maxn][maxn],cost[maxn];
int main(){
	ios::sync_with_stdio(false);
	memset(f,-0x3f,sizeof(f));
	f[0]=0;
	memset(maxx,-0x3f,sizeof(maxx));
	cin>>n>>m>>p;
	for(int i=0;i<n;i++){
		for(int j=1;j<=m;j++){
			cin>>a[i][j];
		}
	}
	for(int i=0;i<n;i++) cin>>cost[i];
	for(int i=m;i>=1;i--){
		for(int j=1;j<=p&&i+j-1<=m;j++){
			for(int k=0;k<n;k++){
				if(j==1) dp[i&1][j][k]=a[k][i];
				else dp[i&1][j][k]=dp[(i+1)&1][j-1][(k+1)%n]+a[k][i];
				maxx[i][j]=max(maxx[i][j],dp[i&1][j][k]-cost[k]);
			}
		}
	}
	for(int i=1;i<=m;i++){
		for(int j=1;j<=i;j++){
			f[i]=max(f[i],f[j-1]+maxx[j][i-j+1]);
		}
	}
	cout<<f[m];
    return 0;
}