Cod sursa(job #1362968)

Utilizator roxana_97Soare Roxana Florentina roxana_97 Data 26 februarie 2015 17:15:33
Problema Ubuntzei Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.38 kb
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#define Nmax 50009
#define oo 2000000000
using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");

int N,M,S,F,d[Nmax],x,y,ante[Nmax],sol,dr[Nmax],K,solp;
bitset < Nmax > inPQ;
struct muchie
{
    int nod,c;
}aux;
vector < muchie > Graph[Nmax];
bool verif[Nmax];
struct cmp
{
    bool operator()(int x, int y)
    {
        if(d[x] > d[y]) return 1;
        return 0;
    }
};
priority_queue < int  , vector < int > , cmp > pq;


void Dijkstra(int S)
{
    for(int i=1;i<=N;++i)d[i]=oo,ante[i]=oo;
    d[S]=ante[S]=0;
    pq.push(S);
    while(!pq.empty())
    {
        int node=pq.top();
        pq.pop();
        inPQ[node]=0;
        for(vector<muchie>::iterator it=Graph[node].begin() ; it!=Graph[node].end() ; ++it)
            if(d[node]+it->c < d[it->nod])
            {
                d[it->nod]=d[node]+it->c;
                ante[it->nod]=node;
                if(inPQ[it->nod]==0) {
                    pq.push(it->nod);
                    inPQ[it->nod]=1;
                }
            }
    }
    for (int i=1; i<=N; ++i)
        if (d[i]==oo)d[i]=0,ante[i]=0;
}


void Rebuild(int S, int node)
{
    verif[node]=0;
    if(node==S){ /*g<<S<<' ';*/ return; }
    Rebuild(S, ante[node]);
    //g<<node<<' ';
}



int main()
{
    f>>N>>M>>K;
    for(int i=1;i<=K;i++)
    {
        f>>dr[i];
        verif[dr[i]]=1;
    }
    for(int i=1;i<=M;++i)

        {
            f>>x>>aux.nod>>aux.c;
            Graph[x].push_back(aux);
            /*
                f>>x>>y>>c;
                aux.nod=y; aux.c=c;
                Graph[x].push_back(aux);
                aux.nod=x; aux.c=c;
                Graph[y].push_back(aux);
            */
        }
        S=1;
    for(int i=1;i<=K;i++)
        if(verif[dr[i]])
        {
            F=dr[i];
            Dijkstra(S);
            //g<<"Drumul de la "<<S<< " la "<<F<<" are lungimea "<<d[F]<<"\n";    	g<<"si este:\n";
            Rebuild(S,F);
            sol+=d[F];
            //g<<'\n'<<sol<<'\n';
            S=dr[i];
        }
        Dijkstra(S);F=N;
        //g<<"Drumul de la "<<S<< " la "<<F<<" are lungimea "<<d[F]<<"\n";    	g<<"si este:\n";
        Rebuild(S,F);
        //g<<'\n';
        sol+=d[F];
    g<<sol<<'\n';
    f.close();g.close();
    return 0;
}