Cod sursa(job #923068)

Utilizator sebii_cSebastian Claici sebii_c Data 22 martie 2013 20:33:45
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.96 kb
#include <cstdio>
#include <cstring>

#include <algorithm>
#include <queue>
#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];
int dist[MAXK][MAXN];

int N, C;
vector<int> G[MAXN];
vector<int> Cst[MAXN];
vector<int> ubun;

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 cost = 0;
    int st = 1;
    for (int i = 0; i < C; ++i) {
        cost += dist[st][ubun[i]];
        st = ubun[i];
    }
    cost += dist[st][N];

    return cost;
}

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);
    }

    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);
    for (int i = 0; i < C; ++i)
        dijkstra(ubun[i]);

    sort(ubun.begin(), ubun.end());

    int min_cost = INF;
    do {
        min_cost = min(min_cost, doit());
    } while (next_permutation(ubun.begin(), ubun.end()));

    printf("%d\n", min_cost);

    return 0;
}