Pagini recente » Cod sursa (job #720921) | Cod sursa (job #1950171) | Cod sursa (job #1386537) | Cod sursa (job #1932943) | Cod sursa (job #1362984)
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#define Nmax 10099
#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) return;
Rebuild(S, ante[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);
}
S=1;
for(int i=1;i<=K;i++)
if(verif[dr[i]])
{
F=dr[i];
Dijkstra(S);
Rebuild(S,F);
sol+=d[F];
S=dr[i];
}
Dijkstra(S);F=N;
Rebuild(S,F);
sol+=d[F];
g<<sol<<'\n';
f.close();g.close();
return 0;
}