poj 1036 Gangsters

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define ll __int64
 5 using namespace std;
 6 
 7 int dp[2][103];
 8 int t[200],p[200],s[200];
 9 int n,k,t1;
10 bool vis[300002];
11 int max1(int a,int b,int c)
12 {
13     return (a>b?a:b)>c?(a>b?a:b):c;
14 }
15 
16 
17 int main()
18 {
19     while(scanf("%d%d%d",&n,&k,&t1)!=EOF)
20     {
21         memset(vis,false,sizeof(vis));
22         for(int i=1; i<=n; i++){
23             scanf("%d",&t[i]);
24             vis[t[i]]=true;
25         }
26         for(int i=1; i<=n; i++)
27             scanf("%d",&p[i]);
28         for(int i=1; i<=n; i++)
29             scanf("%d",&s[i]);
30         memset(dp,0,sizeof(dp));
31         for(int i=0; i<=t1; i++)
32         {
33             for(int j=1; j<=k+1&&j<=i; j++)
34             {
35                 int sum=0;
36                 if(vis[i])
37                 {
38                     for(int c=1; c<=n; c++)
39                     {
40                         if(s[c]==j&&t[c]==i)
41                         {
42                             sum+=p[c];
43                         }
44                     }
45                 }
46                 if(j==1)
47                 {
48                    dp[i&1][j]=max(dp[(i-1)&1][j],dp[(i-1)&1][j+1])+sum;
49                 }
50                 else if(j==k)
51                 dp[i&1][j]=max(dp[(i-1)&1][j],dp[(i-1)&1][j-1])+sum;
52                 else
53                     dp[i&1][j]=max1(dp[(i-1)&1][j],dp[(i-1)&1][j-1],dp[(i-1)&1][j+1])+sum;
54             }
55         }
56         int ans=0;
57         for(int i=1; i<=k+1; i++)
58         {
59             ans=max(ans,dp[t1&1][i]);
60         }
61         printf("%d\n",ans);
62     }
63     return 0;
64 }