类数塔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;
}