poj 1036 Gangsters

http://poj.org/problem?id=1036

题意：N个土匪，伸缩门的范围是K, 时间T， 伸缩门在【0， k】范围内变动，每个单位时间可以不变伸长或者缩短一个单位。给出每个最烦到达的时刻，取得的成就，和肥胖程度。即如果伸缩门的长度和土匪的肥胖程度一样，即得到成就。

状态转移方程：dp[i][j]=max(dp[i-1][j],dp[i-1][j],dp[i-1][j+1])+a[i][j];

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll __int64
using namespace std;

int dp[2][103];
int t[200],p[200],s[200];
int n,k,t1;
bool vis[300002];
int max1(int a,int b,int c)
{
    return (a>b?a:b)>c?(a>b?a:b):c;
}


int main()
{
    while(scanf("%d%d%d",&n,&k,&t1)!=EOF)
    {
        memset(vis,false,sizeof(vis));
        for(int i=1; i<=n; i++){
            scanf("%d",&t[i]);
            vis[t[i]]=true;
        }
        for(int i=1; i<=n; i++)
            scanf("%d",&p[i]);
        for(int i=1; i<=n; i++)
            scanf("%d",&s[i]);
        memset(dp,0,sizeof(dp));
        for(int i=0; i<=t1; i++)
        {
            for(int j=1; j<=k+1&&j<=i; j++)
            {
                int sum=0;
                if(vis[i])
                {
                    for(int c=1; c<=n; c++)
                    {
                        if(s[c]==j&&t[c]==i)
                        {
                            sum+=p[c];
                        }
                    }
                }
                if(j==1)
                {
                   dp[i&1][j]=max(dp[(i-1)&1][j],dp[(i-1)&1][j+1])+sum;
                }
                else if(j==k)
                dp[i&1][j]=max(dp[(i-1)&1][j],dp[(i-1)&1][j-1])+sum;
                else
                    dp[i&1][j]=max1(dp[(i-1)&1][j],dp[(i-1)&1][j-1],dp[(i-1)&1][j+1])+sum;
            }
        }
        int ans=0;
        for(int i=1; i<=k+1; i++)
        {
            ans=max(ans,dp[t1&1][i]);
        }
        printf("%d
",ans);
    }
    return 0;
}