Cod sursa(job #3197770)

Utilizator divadddDavid Curca divaddd Data 27 ianuarie 2024 13:57:14
Problema Pitici Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 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];
vector<pii> v[NMAX];

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

void dfs(int nod){
    if(v[nod].size() == 0){
        dp[nod][1] = 0;
        return ;
    }
    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});
    }
}

int 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));
    dfs(1);
    for(int i = 1; i <= k; i++){
        fout << dp[1][i] << " ";
    }
    return 0;
}