POJ1036 Gangsters DP

题意：一群匪徒要进入一个酒店。酒店的门有k+1个状态，每个匪徒的参数是：进入时间，符合的状态，携带的钱。

酒店的门刚开始状态0，问最多这个酒店能得到的钱数。

思路：

dp数组为DP[T][K].

转移方程dp[i][j]=max(dp[i-1][j],dp[i-1][j-1],dp[i-1][j+1])

因为转移i只跟i-1有关，所以可以用滚动数组dp[2][k].

其实这道题的转移方程很容易想到，只是编程的时候处理边界等细节比较麻烦。

#include<iostream> #include<algorithm> #define max(a,b) (a>b?a:b) using namespace std; const int N=105,K=105,T=30005; int dp[2][K]; struct GAN { int t,p,s; }gan[N]; bool cmp(const GAN &a,const GAN &b) { return a.t<b.t; } int n,k,t; bool Hash[T]; void solve() { for(int i=1;i<=n;i++) Hash[gan[i].t]=true; int w; sort(gan+1,gan+n+1,cmp); for(int i=1;i<=t;i++) { for(int j=0;j<=k&&j<=i;j++)//注意j<=i { w=0; if(Hash[i]) { for(int ii=1;ii<=n;ii++) { if(gan[ii].t==i&&gan[ii].s==j) w+=gan[ii].p; } } if(j==k) dp[i%2][j]=max(dp[(i-1)%2][j-1],dp[(i-1)%2][j]); else if(j==0) dp[i%2][j]=max(dp[(i-1)%2][j+1],dp[(i-1)%2][j]); else dp[i%2][j]=max(dp[(i-1)%2][j-1],max(dp[(i-1)%2][j+1],dp[(i-1)%2][j])); dp[i%2][j]+=w; } } int ans=0; for(int i=0;i<=k;i++) { ans=max(ans,dp[t%2][i]); } printf("%d\n",ans); } int main() { scanf("%d%d%d",&n,&k,&t); for(int j=1;j<=n;j++) { scanf("%d",&gan[j].t); } for(int j=1;j<=n;j++) { scanf("%d",&gan[j].p); } for(int j=1;j<=n;j++) { scanf("%d",&gan[j].s); } solve(); return 0; }