Cod sursa(job #3328787)

Utilizator tonealexandruTone Alexandru tonealexandru Data 10 decembrie 2025 13:11:24
Problema Pitici Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <bits/stdc++.h>

using namespace std;

const int NMAX = 1028;

vector<pair<int, int>> adj[NMAX];
int grad[NMAX];

queue<int> q;
queue<int> top;

vector<int> v[NMAX];
int main() {
  ifstream cin("pitici.in");
  ofstream cout("pitici.out");
  /// WIP
    int n, m, k, a, b, val;
    cin>>n>>m>>k;

    for(int i=0; i<m; i++) {
        cin>>a>>b>>val;
        adj[a].push_back({b, val});
        grad[b] ++;
    }

    for(int i=1; i<=n; i++)
        if(grad[i] == 0)
            q.push(i), top.push(i);

    while(q.empty() == false) {
        int curr = q.front();
        q.pop();

        for(auto x : adj[curr]) {
            grad[x.first]--;
            if(grad[x.first] == 0)
                q.push(curr), top.push(curr);
        }
    }


    while(top.empty() == false) {
        int curr = top.front();
        top.pop();

        sort(v[curr].begin(), v[curr].end());

        for(int i=0; i<k; i++){
            int old_value = v[curr][k];
            for(auto x : adj[curr]) {
                int new_value = old_value + x.second;
                v[x.first].push_back(new_value);
            }
        }
    }

    sort(v[n].begin(), v[n].end());
    for(int i=0; i<k; i++)
      cout<<v[n][i]<<" ";



    return 0;
}