Cod sursa(job #3197777)

Utilizator divadddDavid Curca divaddd Data 27 ianuarie 2024 14:02:40
Problema Pitici Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
using pii = pair<int, int>;
const int NMAX = 2002;
int n,m,k,dp[NMAX][NMAX],vf[NMAX];
vector<pii> v[NMAX];

ifstream fin("pitici.in");
ofstream fout("pitici.out");

void dfs(int nod){
    if(v[nod].size() == 0 || vf[nod] == 1){
        return ;
    }
    vf[nod] = 1;
    set<tuple<int, int, int>> s;
    for(auto [vecin, cost]: v[nod]){
        dfs(vecin);
        s.insert({dp[vecin][1] + cost, vecin, 1});
    }
    for(int i = 1; i <= k; i++){
        auto [drum, vecin, cnt] = *s.begin();
        s.erase(s.begin());
        dp[nod][i] = drum;
        int cost = drum - dp[vecin][cnt];
        s.insert({dp[vecin][cnt+1] + cost, vecin, cnt+1});
    }
}

signed main()
{
    fin >> n >> m >> k;
    for(int i = 1; i <= m; i++){
        int x, y, c;
        fin >> x >> y >> c;
        v[x].push_back({y, c});
    }
    memset(dp, INF, sizeof(dp));
    dp[n][1] = 0;
    dfs(1);
    for(int i = 1; i <= k; i++){
        fout << dp[1][i] << " ";
    }
    return 0;
}