Cod sursa(job #1896904)

Utilizator RaZxKiDDavid Razvan RaZxKiD Data 28 februarie 2017 23:20:20
Problema Amenzi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>

using namespace std;

ifstream in("amenzi.in");
ofstream out("amenzi.out");

int n,m,k,p;

vector<pair<int,int>> L[12005];
int AM[155][3505]; //in nodu i si la momentu j

int DP[155][3505];
//! amenzile maxime daca se afla in nodul N la timpul T,

int tmax;


void read(){
    int x,y,t;
    in>>n>>m>>k>>p;
    for(int i=1;i<=m;i++){
        in>>x>>y>>t;
        L[x].push_back({y,t});
        L[y].push_back({x,t});
    }
    for(int i=1;i<=k;i++){
        in>>x>>y>>t;
        AM[x][y]+=t;
    }
}
void solve(){
    int x,y;
    memset(DP,-1,sizeof(DP));
    DP[1][0]=0;
    for(int t=1;t<=3500;t++){
        for(int nod=1;nod<=n;nod++){
            for(auto vecin : L[nod]){
                if(t-vecin.second>=0)
                    DP[nod][t]=max(DP[nod][t],DP[vecin.first][t-vecin.second]);
            }
            DP[nod][t]=max(DP[nod][t],DP[nod][t-1]);
            if(DP[nod][t]>=0)DP[nod][t]+=AM[nod][t];
        }
    }
    for(int i=1;i<=p;i++){
        in>>x>>y;
        out<<DP[x][y]<<"\n";
    }
}
int main(){
    read();
    solve();
    return 0;
}