Cod sursa(job #1225885)

Utilizator smaraldaSmaranda Dinu smaralda Data 3 septembrie 2014 21:45:02
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.54 kb
#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; ++ i)
        dijkstra(c[i], i);

    dinamica();

    printf("%d\n", d[k + 1][(1 << (k + 2)) - 1]);
    return 0;
}