Pagini recente » Cod sursa (job #750022) | Cod sursa (job #1986606) | Cod sursa (job #913105) | Cod sursa (job #2365959) | Cod sursa (job #2356696)
#include <bits/stdc++.h>
using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");
int n,m,k,p[20],dist[2003],s[2003],must[2003];
int d[2002][2002];
const int INF=0x3f3f3f3f;
int ans=999999999;
set<pair<int, int> >h;
vector<pair<int, int> >v[2003];
void blabla()
{ //g<<p[s[1]]<<' ';
//g<<d[1][2]<<' '<<d[2][n]<<' ';
int D,i;
D=d[1][s[1]];
for(i=2;i<=k;i++) D+=d[s[i-1]][s[i]];
D+=d[s[k]][n];
ans=min(ans,D);
}
bool valid(int pas)
{
if(s[pas]==s[pas-1])return false;
return true;
}
void backtr(int pas)
{
int i;
for(i=1;i<=k;i++)
{
s[pas]=must[i];
if(valid(pas))
{
if(pas==k) blabla();
else backtr(pas+1);
}
}
}
int main()
{
int i,x,y,c,j;
f>>n>>m;
f>>k;
for(i=1;i<=k;i++) f>>must[i];
for(i=1;i<=m;i++)
{
f>>x>>y>>c;
v[x].push_back(make_pair(y,c));
v[y].push_back(make_pair(x,c));
}
//backtr(1);
if(k==0)
{
vector<pair<int, int> >::iterator it;
int nod,to,cost;
///dijkstra
fill(dist+1,dist+n+1,INF);
dist[1]=0;
h.insert(make_pair(0,1));
while(!h.empty())
{
nod=h.begin()->second;
h.erase(h.begin());
for(it=v[nod].begin();it!=v[nod].end();++it)
{
to=it->first; cost=it->second;
if(dist[to]>dist[nod]+cost)
{
if(dist[to]!=INF) h.erase(h.find(make_pair(dist[to],to)));
dist[to]=dist[nod]+cost;
h.insert(make_pair(dist[to],to));
}
}
}
g<<dist[n]<<'\n';
}
else
{
vector<pair<int, int> >::iterator it;
int nod,to,cost;
///dijkstra
for(i=1;i<=n;i++)
{
fill(dist+1,dist+n+1,INF);
dist[i]=0;
h.insert(make_pair(0,i));
while(!h.empty())
{
nod=h.begin()->second;
h.erase(h.begin());
for(it=v[nod].begin();it!=v[nod].end();++it)
{
to=it->first; cost=it->second;
if(dist[to]>dist[nod]+cost)
{
if(dist[to]!=INF) h.erase(h.find(make_pair(dist[to],to)));
dist[to]=dist[nod]+cost;
h.insert(make_pair(dist[to],to));
}
}
}
for(j=1;j<=n;j++)
if(j!=i)
{
d[i][j]=d[j][i]=dist[j];
}
}
backtr(1);
g<<ans<<'\n';
}
f.close();
g.close();
return 0;
}