Cod sursa(job #1579771)

Utilizator VictoriaNevTascau Victoria VictoriaNev Data 25 ianuarie 2016 08:25:54
Problema Amenzi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.19 kb
#include <fstream>
#include <vector>
#include <cstring>
using namespace std;
const int TMAX=3501, NMAX=151;
int cost[TMAX][TMAX], dp[TMAX][TMAX];
int main()
{
    ifstream in("amenzi.in");
    ofstream out("amenzi.out");
    int n, m, k, p, i, j, q, c, u, v, time, node;
    in>>n>>m>>k>>p;

    vector< vector< pair<int, int> > > g(n+1);
    for(i=1; i<=m; i++)
    {
        in>>u>>v>>time;
        g[u].push_back(make_pair(v, time));
        g[v].push_back(make_pair(u, time));
    }
    for(i=1; i<=k; i++)
    {
        in>>u>>v>>c;
        cost[v][u]=c;
    }

    memset(dp, -1, sizeof(dp));
    dp[0][1]=0;

    for(i=1; i<=TMAX; i++)
        for(j=1; j<=n; j++)
        {

            for(q=0; q<g[j].size(); q++)
            {
                node=g[j][q].first;
                time=g[j][q].second;
                if(i-time>=0)
                    dp[i][j]=max(dp[i-time][node], dp[i][j]);

            }
            dp[i][j]=max(dp[i][j], dp[i-1][j]);
            if(dp[i][j]>=0)
                dp[i][j]+=cost[i][j];
        }

    for(i=1; i<=p; i++)
    {
        in>>u>>time;
        if(dp[time][u]<0)
            out<<-1<<'\n';
        else
            out<<dp[time][u]<<'\n';
    }

    return 0;
}