Submission #737728

#	Time	Username	Problem	Language	Result	Execution time	Memory
737728		Ariadna	Autobus (COCI22_autobus)	C++14	15 / 70	82 ms	384 KiB

Main

#include <bits/stdc++.h>
#define ll long long int
 
using namespace std;
 
ll INF = 1e18;
 
int n, m;
vector < vector < pair < int, int > > > adj;
 
ll dijkstra(int s, int f, int k) {
    vector < pair < ll, int > > dist(n, {INF, 0});
    priority_queue < pair < ll, pair < int, int > > > pq;
    dist[s] = make_pair(0, 0);
    pq.push(make_pair(0, make_pair(s, 0)));
    
    while (!pq.empty()) {
        ll d = -pq.top().first;
        int u = pq.top().second.first;
        int l = pq.top().second.second;
        pq.pop();
        if (l == k) continue;
        if (dist[u].first != d && l >= dist[u].second) continue;
        for (pair < int, int > vecino : adj[u]) {
            int v = vecino.first;
            int w = vecino.second;
            if (dist[v].first > d + w) {
                dist[v] = make_pair(d + w, l + 1);
                pq.push(make_pair(-dist[v].first, make_pair(v, l + 1)));
            }
        }
    }
    if (dist[f].first == INF) return -1;
    return dist[f].first;
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    
    cin >> n >> m;
    adj = vector < vector < pair < int, int > > >(n);
    while (m--) {
        int a, b, t;
        cin >> a >> b >> t;
        --a; --b;
        adj[a].push_back(make_pair(b, t));
    }
    int k, q;
    cin >> k >> q;
    while (q--) {
        int a, b;
        cin >> a >> b;
        cout << dijkstra(a - 1, b - 1, k) << '\n';
    }
    
    return 0;
}

• Subtask 1: 15 / 15
• Subtask 2: 0 / 15
• Subtask 3: 0 / 25
• Subtask 4: 0 / 15