Pagini recente » Cod sursa (job #847654) | Cod sursa (job #148208) | Cod sursa (job #3201393) | Cod sursa (job #3039904) | Cod sursa (job #2323068)
#include <bits/stdc++.h>
#define pii pair<int,int>
#define piii pair<pii,int>
#define INF INT_MAX
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
vector<pair<int,int> >nod[2005];
int n,m,k,x,y,c;
int pretin[20];
int val[2005];
int dp[2005][2005];
vector<int>elem[2005],fri;
bitset<32800>ap[2005];
priority_queue<pii,vector<pii >,greater<pii > >pq;
void dijkstra(int node)
{
pq.push({0,node});
while(!pq.empty())
{
int pos=pq.top().second;
int v=pq.top().first;
pq.pop();
if(val[pos]!=INF) continue;
val[pos]=v;
for(int i=0;i<nod[pos].size();i++)
if(val[nod[pos][i].first]==INF)
pq.push({nod[pos][i].second+v,nod[pos][i].first});
}
}
void get_ans()
{
priority_queue<piii,vector<piii >,greater<piii > >q;
q.push({{0,0},1});
while(!q.empty())
{
int pos=q.top().second;
int cst=q.top().first.first;
int v=q.top().first.second;
q.pop();
if(ap[pos][v]) continue;
ap[pos][v]=1;
if(pos==n && v==(1<<k)-1)
{
fout<<cst;
return;
}
for(int i=0;i<fri.size();i++)
{
if(pos==fri[i]) continue;
int counter=v;
if(val[fri[i]]!=-1 && (v&(val[fri[i]]))==0)
counter+=val[fri[i]];
if(ap[fri[i]][counter]) continue;
q.push({{cst+dp[fri[i]][pos],counter},fri[i]});
}
}
}
int main()
{
fin>>n>>m>>k;
for(int i=0;i<k;i++)
fin>>pretin[i];
for(int i=1;i<=m;i++)
{
fin>>x>>y>>c;
nod[x].push_back({y,c});
nod[y].push_back({x,c});
}
for(int j=0;j<=n;j++) val[j]=INF;
dijkstra(1);
for(int i=0;i<k;i++)
{
dp[1][pretin[i]]=val[pretin[i]];
dp[pretin[i]][1]=val[pretin[i]];
}
dp[1][n]=val[n];
dp[n][1]=val[n];
for(int i=0;i<k;i++)
{
for(int j=0;j<=n;j++) val[j]=INF;
dijkstra(pretin[i]);
for(int j=0;j<k;j++)
dp[pretin[i]][pretin[j]]=val[pretin[j]];
dp[pretin[i]][n]=val[n];
dp[n][pretin[i]]=val[n];
}
for(int j=0;j<=n;j++) val[j]=-1;
fri.push_back(1);
for(int i=0;i<k;i++)
{
val[pretin[i]]=1<<i;
fri.push_back(pretin[i]);
}
fri.push_back(n);
get_ans();
return 0;
}