Pagini recente » Cod sursa (job #1975536) | Cod sursa (job #216244) | Cod sursa (job #370983) | Cod sursa (job #29406) | Cod sursa (job #1225886)
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define ITERATOR vector <NODE>::iterator
const int NMAX = 2010, KMAX = 18, CONF = 1 << 18, INF = 2e9;
struct NODE {
int node, cost;
NODE (int node = 0, int cost = 0) {
this->node = node;
this->cost = cost;
}
const bool operator < (const NODE &other) const{
return cost > other.cost;
}
};
vector <NODE> pre[NMAX], post[NMAX];
int n, k, dmin[NMAX], c[NMAX], d[KMAX][CONF], dist[KMAX][KMAX];
bool vis[NMAX];
priority_queue <NODE> heap;
void initDijkstra() {
for(int i = 1; i <= n; ++ i) {
dmin[i] = INF;
vis[i] = 0;
}
}
void initDinamica () {
for(int i = 0; i <= k + 1; ++ i)
for(int j = 0; j < (1 << (k + 2)); ++ j)
d[i][j] = INF;
d[0][0] = 0;
}
void updateDmin (int node) {
for(ITERATOR it = pre[node].begin(); it != pre[node].end(); ++ it)
if(dmin[node] + it->cost < dmin[it->node]) {
dmin[it->node] = dmin[node] + it->cost;
heap.push(NODE(it->node, dmin[it->node]));
}
}
void dijkstra (int start, int ind) {
ITERATOR it;
int node, i;
initDijkstra();
dmin[start] = 0;
updateDmin(start);
while(!heap.empty()) {
node = heap.top().node;
updateDmin(node);
vis[node] = 1;
while(!heap.empty() && vis[heap.top().node])
heap.pop();
}
for(i = 0; i <= k + 1; ++ i)
dist[ind][i] = dist[i][ind] = dmin[c[i]];
}
void dinamica() {
int node, conf, add;
initDinamica();
for(node = 0; node <= k + 1; ++ node)
for(conf = 0; conf < (1 << (k + 2)); ++ conf)
if(d[node][conf] != INF)
for(add = 0; add <= k + 1; ++ add)
if((conf & (1 << add)) == 0)
d[add][conf | (1 << add)] = min(d[add][conf | (1 << add)], d[node][conf] + dist[node][add]);
}
int main() {
freopen("ubuntzei.in", "r", stdin);
freopen("ubuntzei.out", "w", stdout);
int m, i, x, y, cost;
scanf("%d%d%d", &n, &m, &k);
for(i = 1; i <= k; ++ i)
scanf("%d", &c[i]);
c[0] = 1;
c[k + 1] = n;
for(i = 1; i <= m; ++ i) {
scanf("%d%d%d", &x, &y, &cost);
pre[x].push_back(NODE(y, cost));
pre[y].push_back(NODE(x, cost));
}
for(i = 0; i <= k + 1; ++ i)
dijkstra(c[i], i);
dinamica();
printf("%d\n", d[k + 1][(1 << (k + 2)) - 1]);
return 0;
}