Pagini recente » Cod sursa (job #1919734) | Cod sursa (job #602201) | Cod sursa (job #792085) | Cod sursa (job #1633377) | Cod sursa (job #1111370)
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
#include <queue>
using namespace std;
#define MAXN 2001
#define MAXK 15
#define inf 0x3f3f3f3f
typedef vector< pair<int, int> >::iterator iter;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");
int n, m, k;
int c[MAXK];
vector< pair<int, int> > G[MAXN];
int src[MAXN], d[MAXK][MAXN];
int cost[MAXN][1 << MAXK]; // cost[j][i] = costul minim al unui drum care incepe
// in nodul 1, trece prin toate nodurile multimea i si se termina in j.
bitset<MAXN> viz;
priority_queue< pair<int, int> > pq;
void dijkstra(int nod, int *dist)
{
for (int i = 1; i <= n; i++) {
dist[i] = inf;
}
viz.reset();
dist[nod] = 0;
pq.push(make_pair(0, nod));
while (!pq.empty()) {
int nd = pq.top().second;
pq.pop();
if (viz[nd] == true) {
continue;
}
viz[nd] = true;
for (iter it = G[nd].begin(); it != G[nd].end(); it++) {
if (dist[it->first] > dist[nd] + it->second) {
dist[it->first] = dist[nd] + it->second;
pq.push(make_pair(-dist[it->first], it->first));
}
}
}
}
int hamilton()
{
for (int i = 1; i < (1 << k); i++) {
bool cont = false;
for (int j = 0; j < k; j++) {
if (i == (1 << j)) {
cost[j][i] = src[c[j]];
cont = true;
break;
}
}
if (cont == true) {
continue;
}
for (int j = 0; j < k; j++) {
cost[j][i] = inf;
if (i & (1 << j)) {
for (int l = 0; l < k; l++) {
if (l != j && (i & (1 << l))) {
cost[j][i] = min(cost[j][i], cost[l][i - (1 << j)] + d[l][c[j]]);
}
}
}
}
}
int res = inf;
for (int i = 0; i < k; i++) {
res = min(res, cost[i][(1 << k) - 1] + d[i][n]);
}
return res;
}
int main()
{
f >> n >> m >> k;
for (int i = 0; i < k; i++) {
f >> c[i];
}
for (int i = 0; i < m; i++) {
int x, y, z;
f >> x >> y >> z;
G[x].push_back(make_pair(y, z));
G[y].push_back(make_pair(x, z));
}
dijkstra(1, src);
if (k == 0) {
g << src[n] << endl;
} else {
for (int i = 0; i < k; i++) {
dijkstra(c[i], d[i]);
}
g << hamilton() << endl;
}
f.close();
g.close();
return 0;
}