Cod sursa(job #2858040)

Utilizator CelestinNegraru Celestin Celestin Data 26 februarie 2022 21:00:09
Problema Pitici Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.84 kb
#include <fstream>
#include <vector>
#include <set>
#define maxi 1050
using namespace std;
ifstream f("pitici.in");
ofstream g("pitici.out");
multiset<int> s;
multiset<int>::iterator I;
vector<pair<int,int>> G[maxi],GT[maxi];
int n,m,a,b,c,F[maxi],l,poz[maxi],dproads[maxi][maxi],dpnr[maxi],k;
bool viz[maxi];
void read()
{
    f>>n>>m>>k;
    for(int i=1;i<=m;i++)
    {
        f>>a>>b>>c;
        G[a].push_back(make_pair(b,c));
        GT[b].push_back(make_pair(a,c));
    }
    f.close();
    return;
}
void dfs(int x)
{
   viz[x]=true;
   for(auto a:G[x])
    if(!viz[a.first])
     dfs(a.first);
   F[++l]=x;
   poz[x]=n+1-l;
}
void solve()
{
    read();
    for(int i=1;i<=n;i++)
        if(!viz[i])
          dfs(i);
    dpnr[1]=1;
    for(int i=l;i>=1;i--)
    {
        int nod=F[i];
        s.clear();
        for(auto a:GT[nod])
        {
            int vecin=a.first;
            int cost=a.second;
            if(poz[vecin]<poz[nod])
            {
                for(int j=1;j<=dpnr[vecin];j++)
                {
                    if(s.size()<k)
                    {
                        s.insert(dproads[vecin][j]+cost);
                    }
                    else{
                        I=s.end();
                        I--;
                        if(dproads[vecin][j]+cost<*I)
                        {
                            s.erase(I);
                            s.insert(dproads[vecin][j]+cost);
                        }
                    }
                }
                dpnr[nod]=s.size();
                int p=0;
                for(auto r:s)
                {
                    dproads[nod][++p]=r;
                }
            }
        }
    }
    for(int i=1;i<=k;i++)
      g<<dproads[n][i]<<" ";
}
int main()
{
    solve();
    return 0;
}