Pagini recente » Cod sursa (job #2579874) | Cod sursa (job #2401474) | Cod sursa (job #1619281) | Cod sursa (job #1254513) | Cod sursa (job #2562958)
#include <fstream>
#include <vector>
#include <queue>
#define pinf 1000000007
#define f first
#define s second
using namespace std;
ifstream fin ("ubuntzei.in");
ofstream fout ("ubuntzei.out");
int n, m, k, pr[20], x, y, l, a[20][2005], d[2005], viz[2005], pd[32800][20], nr, aux, p, rez;
vector <pair <int, int> > v[2005];
priority_queue < pair <int, int>, vector < pair <int, int> >, greater < pair <int, int> > > pq;
pair <int, int> mini;
void dijkstra (int start)
{
int i;
for (i=1; i<=n; i++)
{
d[i]=pinf;
viz[i]=0;
}
viz[start]=1;
d[start]=0;
for (auto it: v[start])
{
d[it.s]=it.f;
pq.push(make_pair(d[it.s], it.s));
}
while (pq.empty()==0)
{
mini=pq.top();
pq.pop();
if (viz[mini.s]==0)
{
viz[mini.s]=1;
for (auto it: v[mini.s])
{
if (viz[it.s]==0)
{
if (d[mini.s]+it.f<d[it.s])
{
d[it.s]=d[mini.s]+it.f;
pq.push(make_pair(d[it.s], it.s));
}
}
}
}
}
}
int main()
{
int i, j, r;
fin>>n>>m>>k;
for (i=0; i<k; i++)
{
fin>>pr[i];
}
for (i=1; i<=m; i++)
{
fin>>x>>y>>l;
v[x].push_back(make_pair(l, y));
v[y].push_back(make_pair(l, x));
}
if (k==0)
{
dijkstra(1);
fout<<d[n];
}
else
{
for (i=0; i<k; i++)
{
dijkstra(pr[i]);
for (j=1; j<=n; j++)
{
a[i][j]=d[j];
}
}
nr=(1<<k)-1;
for (i=0; i<=nr; i++)
{
for (j=0; j<k; j++)
pd[i][j]=pinf;
}
for (i=0; i<k; i++)
{
pd[1<<i][i]=a[i][1];
}
for (i=1; i<=nr; i++)
{
for (j=0; j<k; j++)
{
p=1<<j;
if ((p&i)>0)
{
aux=i-p;
for (r=0; r<k; r++)
{
if (pd[aux][r]+a[r][pr[j]]<pd[i][j])
{
pd[i][j]=pd[aux][r]+a[r][pr[j]];
}
}
}
}
}
rez=pinf;
for (j=0; j<k; j++)
{
if (pd[nr][j]+a[j][n]<rez)
rez=pd[nr][j]+a[j][n];
}
fout<<rez;
}
fin.close();
fout.close();
return 0;
}