Cod sursa(job #3336567)

Utilizator todoranstefaniaTodoran Stefania todoranstefania Data 24 ianuarie 2026 22:02:55
Problema Ubuntzei Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.97 kb
#include <iostream>
#include <vector>
#include <algorithm>
#include <fstream>
#include <queue>
#define inf 1000009
#define pii pair<int, int>
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
vector<pair<int, int>>g[2005];
priority_queue<pii, vector<pii>, greater<pii>>pq;
int n, m, x, y, z, k;
int c[2005];
int d[2005];
void dijkstra(int start)
{
    for(int i=1; i<=n; i++)
        d[i]=inf;
    d[start]=0;
    pq.push({0, start});
    while(!pq.empty())
    {
        int nod_curent=pq.top().second;
        int dist=pq.top().first;
        pq.pop();
        if(dist>d[nod_curent])
            continue;
        for(auto x: g[nod_curent])
        {
            if(d[x.second]>d[nod_curent]+x.first)
            {
                d[x.second]=d[nod_curent]+x.first;
                pq.push({d[x.second], x.second});
            }
        }
    }
}
int main()
{
    fin>>n>>m>>k;
    for(int i=1; i<=k; i++)
        fin>>c[i];
    sort(c+1,c+k+1);
    for(int i=1; i<=m; i++)
    {
        fin>>x>>y>>z;
        g[x].push_back({z,y});
    }
    if(k==0){
        dijkstra(1);
        fout<<d[n];
    }
    else{
    int mini=100005;
    int sum=0;
    int nod=1;
    while(nod!=n)
    {
        dijkstra(nod);
        int ok=0;
        mini=100005;
        for(int i=1; i<=k; i++)
        {
            if(c[i]!=-1)
                ok=1;
        }
        if(ok)
        {
            for(int i=1; i<=k; i++)
            {
                if(c[i]>0 && c[i]!=n)
                {
                    if(d[c[i]]<mini)
                    {
                        mini=d[i];
                        nod=i;
                    }
                }
            }
            sum+=mini;
            c[nod]=-1;
        }
        else
        {
            sum+=d[n];
            nod=n;
        }

    }
    cout<<sum<<"\n\n";
    for(int i=1; i<=n; i++)
        cout<<d[i]<<" ";
    }
    return 0;
}