Pagini recente » Cod sursa (job #3317383) | Cod sursa (job #3312736) | Istoria paginii utilizator/leonaraging | Istoria paginii utilizator/feardotcom | Cod sursa (job #3336572)
#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});
g[y].push_back({z,x});
}
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;
}