POJ 1036 Gangsters

N gangsters are going to a restaurant. The i-th gangster comes at the time Ti and has the prosperity Pi. The door of the restaurant has K+1 states of openness expressed by the integers in the range [0, K]. The state of openness can change by one in one unit of time; i.e. it either opens by one, closes by one or remains the same. At the initial moment of time the door is closed (state 0). The i-th gangster enters the restaurant only if the door is opened specially for him, i.e. when the state of openness coincides with his stoutness Si. If at the moment of time when the gangster comes to the restaurant the state of openness is not equal to his stoutness, then the gangster goes away and never returns.
The restaurant works in the interval of time [0, T].
The goal is to gather the gangsters with the maximal total prosperity in the restaurant by opening and closing the door appropriately.

?The first line of the input file contains the values N, K, and T, separated by spaces. (1 <= N <= 100 ,1 <= K <= 100 ,0 <= T <= 30000 )
?The second line of the input file contains the moments of time when gangsters come to the restaurant T1, T2, ..., TN, separated by spaces. ( 0 <= Ti <= T for i = 1, 2, ..., N)
?The third line of the input file contains the values of the prosperity of gangsters P1, P2, ..., PN, separated by spaces. ( 0 <= Pi <= 300 for i = 1, 2, ..., N)
?The forth line of the input file contains the values of the stoutness of gangsters S1, S2, ..., SN, separated by spaces. ( 1 <= Si <= K for i = 1, 2, ..., N)
All values in the input file are integers.

Print to the output file the single integer ?the maximal sum of prosperity of gangsters in the restaurant. In case when no gangster can enter the restaurant the output should be 0.

4 10 20
10 16 8 16
10 11 15 1
10 7 1 8

26

CODE:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define REP(i, s, n) for(int i = s; i <= n; i ++)
#define REP_(i, s, n) for(int i = n; i >= s; i --)
#define MAX_N 100 + 10

using namespace std;

int n, k, t;
struct node{
    int T, P, S;
}a[MAX_N];
int F[MAX_N];

bool cmp(node a,node b){ return a.T < b.T; }
int main(){
    scanf("%d%d%d", &n, &k, &t);
    REP(i, 1, n) scanf("%d", &a[i].T);
    REP(i, 1, n) scanf("%d", &a[i].P);
    REP(i, 1, n) scanf("%d", &a[i].S);
    
    sort(a + 1, a + n + 1, cmp);
    REP(i, 1, n){
        if(a[i].T >= a[i].S) F[i] = a[i].P;
        else {F[i] = -1; continue; }
        REP(j, 1, i - 1) 
            if(abs(a[i].S - a[j].S) <= abs(a[i].T - a[j].T)) F[i] = max(F[i], F[j] + a[i].P);
    }
    
    int ans = 0;
    REP(i, 1, n) ans = max(ans, F[i]);
    printf("%d\n", ans);
    return 0;
}