Cod sursa(job #2407329)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 16 aprilie 2019 19:33:43
Problema Amenzi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 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)
        {
            viz[i][j] |= viz[i-1][j];
            if(!viz[i][j])
                continue;
            mx[i][j] = max(mx[i][j], mx[i-1][j]) + xtr[i][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 > 3500)
                    continue;
                viz[i+timp][vec] = 1;
                mx[i+timp][vec] = max(mx[i+timp][vec], mx[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;
}