Cod sursa(job #2407390)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 16 aprilie 2019 20:25:10
Problema Amenzi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.36 kb

#include<bits/stdc++.h>
#pragma gcc optimize("O3")
using namespace std;
ifstream f("amenzi.in");
ofstream g("amenzi.out");
int n, m, k, p;
int xtr[3502][152], mx[3502][152];
vector<pair<int, int> >v[152];
bool viz[3502][152];
int main()
{
    f >> n >> m >> k >> p;
    for(int i = 1; i <= m; ++i)
    {
        int a, b, c;
        f >> a >> b >> c;
        v[a].push_back({b, c});
        v[b].push_back({a, c});
    }
    for(int i = 1; i <= k; ++i)
    {
        int a, b, c;
        f >> a >> b >> c;
        ++b;
        xtr[b][a] += c;
    }
    viz[1][1] = 1;
    for(int i = 1; i <= 3501; ++i)
        for(int j = 1; j <= n; ++j)
        {
            for(int q = 0; q < v[j].size(); ++q)
            {
                int vec = v[j][q].first;
                int timp = v[j][q].second;
                if(i-timp <= 0 || !viz[i-timp][vec])
                    continue;
                viz[i][j] = 1;
                mx[i][j] = max(mx[i][j], mx[i-timp][vec]);
            }
            viz[i][j] |= viz[i-1][j];
            if(viz[i][j])
                mx[i][j] = max(mx[i][j], mx[i-1][j]) + xtr[i][j];
        }
    for(int i = 1; i <= p; ++i)
    {
        int maax = -1;
        int a, b, c;
        f >> a >> b;
        ++b;
        if(viz[b][a])
            maax = mx[b][a];
        g << maax << '\n';
    }
    return 0;
}