poj1036 Gangsters

类数塔DP，简单的存储所有状态会超内存，这里采用滚动数组，另设一个数组mark[]用于记录输入数据中出现的状态，类似于状态压缩。

状态转移方程：dp[i][j]=max(dp[i-1][j-1],dp[i-1][j],dp[i-1][j+1])+p[i][j]         //i是时间，j是门的打开程度。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node{
	int t;
	int p;
	int s;
}gangster[110];
int mark[30010];
int dp[3][110];
bool cmp(node a,node b){
	if(a.t<b.t) return true;
	else return false;
}
int maxi(int a,int b){
	if(a>b) return a;
	else return b;
}
int main(){
	int n,k,t;
	int i,j;
	int w,pi,mabi;
	cin>>n>>k>>t;
	for(i=1;i<=n;i++){
		cin>>gangster[i].t;
		mark[gangster[i].t]=1;
	}
	for(i=1;i<=n;i++)
		cin>>gangster[i].p;
	for(i=1;i<=n;i++)
		cin>>gangster[i].s;
	sort(gangster+1,gangster+n+1,cmp);
	mabi=0;
	for(i=0;i<=t;i++)
		for(j=0;j<=i&&j<=k;j++){
			w=0;
			dp[i%2][j]=0;
			if(mark[i]==1){
				for(pi=1;pi<=n;pi++){
					if(gangster[pi].t==i&&gangster[pi].s==j)
						w+=gangster[pi].p;
				}
			}
			if(j==0)
				dp[i%2][j]=maxi(dp[1-i%2][j],dp[1-i%2][j+1]);
			else if(j==k)
				dp[i%2][j]=maxi(dp[1-i%2][j-1],dp[1-i%2][j]);
			else
				dp[i%2][j]=maxi(maxi(dp[1-i%2][j-1],dp[1-i%2][j]),dp[1-i%2][j+1]);
			dp[i%2][j]+=w;
			if(dp[i%2][j]>mabi)
				mabi=dp[i%2][j];
		}
	cout<<mabi<<endl;
	return 0;
}