Pagini recente » Cod sursa (job #3000108) | Cod sursa (job #2781304) | Cod sursa (job #265198) | Cod sursa (job #1350011) | Cod sursa (job #923411)
Cod sursa(job #923411)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
priority_queue<pair<int, int>, vector<pair<int, int> >,
greater<pair<int, int> > > pq;
const int INF = 0x3f3f3f3f;
const int MAXN = 2001;
const int MAXK = 16;
bool vis[MAXN];
map<int, vector<int> > dist;
int N, C;
vector<int> G[MAXN];
vector<int> Cst[MAXN];
vector<int> ubun;
map<int, vector<int> > dp;
void dijkstra(int s)
{
memset(vis, false, sizeof(vis));
for (int i = 1; i <= N; ++i)
dist[s][i] = INF;
pq.push(make_pair(0, s));
dist[s][s] = 0;
while (!pq.empty()) {
int node = pq.top().second;
int cost = pq.top().first;
pq.pop();
if (vis[node])
continue;
vis[node] = true;
for (int i = 0; i < G[node].size(); ++i) {
int next = G[node][i];
if (cost + Cst[node][i] < dist[s][next]) {
dist[s][next] = cost + Cst[node][i];
pq.push(make_pair(dist[s][next], next));
}
}
}
}
int doit(int i, set<int> s)
{
s.erase(i);
if (s.empty())
return dist[1][i];
int best_sum = INF;
set<int>::iterator it;
for (it = s.begin(); it != s.end(); ++it)
best_sum = min(best_sum, doit(*it, s) + dist[i][*it]);
return best_sum;
}
int main()
{
freopen("ubuntzei.in", "r", stdin);
freopen("ubuntzei.out", "w", stdout);
int M;
scanf("%d %d", &N, &M);
scanf("%d", &C);
for (int i = 0; i < C; ++i) {
int x;
scanf("%d", &x);
ubun.push_back(x);
dist[x] = vector<int>(N + 1);
}
dist[1] = vector<int>(N + 1);
dist[N] = vector<int>(N + 1);
for (int i = 0; i < M; ++i) {
int x, y, c;
scanf("%d %d %d", &x, &y, &c);
G[x].push_back(y);
Cst[x].push_back(c);
G[y].push_back(x);
Cst[y].push_back(c);
}
dijkstra(1);
dijkstra(N);
for (int i = 0; i < C; ++i)
dijkstra(ubun[i]);
set<int> s;
for (int i = 0; i < C; ++i)
s.insert(ubun[i]);
printf("%d\n", doit(N, s));
return 0;
}