Pagini recente » Cod sursa (job #1207188) | Cod sursa (job #1905083) | Cod sursa (job #677253) | Cod sursa (job #2235323) | Cod sursa (job #2685501)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <list>
using namespace std;
const int inf=1<<30;
int dijkstra(int n,int k,const vector<vector<pair<int,int> > >&adj,vector<int> &h)
{
vector<vector<int> >dist(n+1,vector<int>((1<<k),inf));
vector<vector<bool> >visit(n+1,vector<bool>((1<<k),false));
//priority_queue<pair<int,pair<int,int> > >q;
priority_queue<pair<int,pair<int,int> >,vector<pair<int,pair<int,int> > >,greater<pair<int,pair<int,int> > > >q;
q.push({0,{1,0}});
dist[1][0]=0;
while(!q.empty())
{
auto cd=q.top().first;
auto cn=q.top().second.first;
auto cs=q.top().second.second;
q.pop();
visit[cn][cs]=true;
if(cd==dist[cn][cs])
{
if((h[cn]) &&((cs&h[cn])==0)){
auto newset=cs|h[cn];
if((!visit[cn][newset]) && (dist[cn][newset]>dist[cn][cs]))
{
dist[cn][newset]=dist[cn][cs];
q.push({dist[cn][newset],{cn,newset}});
}
}
}
for(auto p :adj[cn])
{
int v=p.first;
int e=p.second;
if((!visit[v][cs]) && dist[v][cs]>e+dist[cn][cs])
{
dist[v][cs]=e+dist[cn][cs];
q.push({dist[v][cs],{v,cs}});
}
}
}
return dist[n][(1<<k)-1];
}
int main()
{
ifstream be("ubuntzei.in");
ofstream ki("ubuntzei.out");
int n,m,k;
be>>n>>m>>k;
vector<int>h(n+1,0);
for(int i=0;i<k;i++)
{
int f;
be>>f;
h[f]=1<<i;
}
vector<vector<pair<int,int> > >adj(n+1);
for(int i=0;i<m;i++)
{
int x,y,cost;
be>>x>>y>>cost;
adj[x].push_back({y,cost});
adj[y].push_back({x,cost});
}
ki<<dijkstra(n,k,adj,h);
return 0;
}