Cod sursa(job #2356573)

Utilizator TudorFinaruTudor Cristian Finaru TudorFinaru Data 26 februarie 2019 19:52:00
Problema Ubuntzei Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");

int n,m,k,p[20],dist[2003];
const int INF=0x3f3f3f3f;

set<pair<int, int> >h;
vector<pair<int, int> >v[2003];

int main()
{
    int i,x,y,c;
    f>>n>>m;
    f>>k;
    for(i=1;i<=k;i++) f>>p[i];
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>c;
        v[x].push_back(make_pair(y,c));
        v[y].push_back(make_pair(x,c));
    }
    if(k==0)
    {
        vector<pair<int, int> >::iterator it;
        int nod,to,cost;
        ///dijkstra
        fill(dist+1,dist+n+1,INF);
        dist[1]=0;
        h.insert(make_pair(0,1));
        while(!h.empty())
        {
            nod=h.begin()->second;
            h.erase(h.begin());
            for(it=v[nod].begin();it!=v[nod].end();++it)
            {
                to=it->first; cost=it->second;
                if(dist[to]>dist[nod]+cost)
                {
                    if(dist[to]!=INF) h.erase(h.find(make_pair(dist[to],to)));
                    dist[to]=dist[nod]+cost;
                    h.insert(make_pair(dist[to],to));
                }
            }
        }
        g<<dist[n]<<'\n';
    }
    f.close();
    g.close();
    return 0;
}