Pagini recente » Cod sursa (job #2880881) | Cod sursa (job #797415) | Cod sursa (job #2374693) | Cod sursa (job #985153) | Cod sursa (job #1307597)
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
const int NMAX = 2000;
const int KMAX = 15;
const int INF = 2000000000;
struct drum
{
int dest;
int lung;
drum (int dest = 0, int lung = 0)
{
this->dest = dest;
this->lung = lung;
}
};
int n, m, k, tata;
int dist[KMAX + 1][NMAX + 1], oras[NMAX + 1], cost[(1 << (KMAX + 1))][NMAX + 1];
vector <drum> v[NMAX + 1];
struct cmp
{
bool operator () (const drum &a, const drum &b)
{
return a.lung > b.lung;
}
};
priority_queue <drum, vector <drum>, cmp> hp;
void dijkstra (int nod)
{
int fiu;
hp.push (drum (nod, 0));
while (!hp.empty ())
{
nod = hp.top ().dest;
hp.pop ();
for (int i = 0; i < v[nod].size (); ++i)
{
fiu = v[nod][i].dest;
if (!dist[tata][fiu] || dist[tata][fiu] > dist[tata][nod] + v[nod][i].lung)
{
dist[tata][fiu] = dist[tata][nod] + v[nod][i].lung;
hp.push (drum (fiu, dist[tata][fiu]));
}
}
}
}
int din (int stare, int ultim)
{
if (cost[stare][ultim])
return cost[stare][ultim];
if (stare == (1 << ultim))
return dist[k][oras[ultim]];
int sol = INF, x = stare - ( 1 << ultim);
for (int i = 0; i < k; ++i)
{
if (x & (1 << i))
{
int aux = din(x, i) + dist[i][oras[ultim]];
if (sol > aux)
sol = aux;
}
}
cost[stare][ultim] = sol;
return sol;
}
int main ()
{
freopen ("ubuntzei.in", "r", stdin);
freopen ("ubuntzei.out", "w", stdout);
int x, y, c;
scanf ("%d%d%d", &n, &m, &k);
for (int i = 0; i < k; ++i)
scanf ("%d", &oras[i]);
for (int i = 0; i < m; ++i)
{
scanf ("%d%d%d", &x, &y, &c);
v[x].push_back (drum (y, c));
v[y].push_back (drum (x, c));
}
for (int i = 0; i < k; ++i)
{
tata = i;
dijkstra (oras[i]);
}
tata = k;
dijkstra (1);
int sol = INF;
if (k == 0)
sol = dist[0][n];
for (int i = 0; i < k; ++i)
{
int aux = din((1 << k) - 1, i) + dist[i][n];
if( aux < sol)
sol = aux;
}
printf ("%d\n", sol);
return 0;
}