Cod sursa(job #2407384)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 16 aprilie 2019 20:21:34
Problema Amenzi Scor 80
Compilator cpp-32 Status done
Runda Arhiva de probleme Marime 1.42 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])
            {
                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;
}