Cod sursa(job #1118857)

Utilizator supermitelArdelean Razvan Mitel supermitel Data 24 februarie 2014 13:37:32
Problema Ubuntzei Scor 45
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.87 kb
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#define INF 9999999

using namespace std;

struct vecin
{
    int nod, d;
    vecin(int nod = 0, int d = 0): nod(nod), d(d)
    {

    }
};

vector<vecin> vec[2011];
int cost[20][20];
int best[2011];

struct cmp
{
    bool operator()(const int & a, const int &b)
    {
        return best[a] > best[b];
    }
};

int n, k;
int mat[20][132000];
int prieten[2011];
int pr[20];

void citire()
{
    int m, x, y, c;
    scanf("%d%d", &n, &m);
    scanf("%d", &k);
    pr[0] = 1;
    for(int i = 1; i <= k; i++)
    {
        scanf("%d", &pr[i]);
        prieten[pr[i]] = 1;
    }

    k++;
    pr[k++] = n;
    for(int i = 0; i < m; i++)
    {
        scanf("%d%d%d", &x, &y, &c);
        vec[x].push_back(vecin(y, c));
        vec[y].push_back(vecin(x, c));
    }
    for(int i = 0; i < k; i++)
    {
        for(int j = 1; j < 1<<k; j++)
            mat[i][j] = INF;
    }
}

priority_queue<int, vector<int>, cmp> q;

void dijkstra(int x)
{
    int start = pr[x];
    for(int i = 0; i <= n; i++)
        best[i] = INF;
    int nou, crt;
    int c;
    best[start] = 0;
    q.push(start);

    while(!q.empty())
    {
        crt = q.top();
        q.pop();
        for(int i = 0; i < vec[crt].size(); i++)
        {
            nou = vec[crt][i].nod;
            c = best[crt] + vec[crt][i].d;
            if(c < best[nou])
            {
                best[nou] = c;
                q.push(nou);
            }
        }
    }
    for(int i = 0; i < k; i++)
        cost[x][i] = cost[i][x] = best[pr[i]];
}

void calc_cost()
{
    for(int i = 0; i < k-1; i++)
        dijkstra(i);
}

void debug_cost()
{
    for(int i = 0; i < k; i++)
    {
        for(int j = 0; j < k; j++)
            cerr<<cost[i][j]<<'\t';
        cerr<<'\n';
    }
    cerr<<endl;
}

void debug_dinam()
{
    for(int i = 0; i < k; i++)
    {
        for(int j = 0; j < 1<<k; j++)
            if(mat[i][j] == INF)
                cout<<-1<<'\t';
            else
                cout<<mat[i][j]<<'\t';
        cout<<'\n';
    }
    cout<<endl;
}

int dinamica(int i, int s)
{
    if(i < 0 || s < 0)
        return INF;
    if(mat[i][s] != INF)
        return mat[i][s];
    if(!(s & (1 << i)))
        return mat[i][s] = INF-1;
    for(int j = 0; j < k; j++)
        if(j != i && (s & (1<<j)))
            mat[i][s] = min(mat[i][s], dinamica(j, s & ~(1 << i)) + cost[i][j]);
    return mat[i][s];
}

int main()
{
    freopen("ubuntzei.in", "r", stdin);
    freopen("ubuntzei.out", "w", stdout);
    citire();
    calc_cost();
    //debug_cost();
    for(int i =0; i < k; i++)
        mat[i][(1<<i)+1] = cost[0][i];
    printf("%d\n", dinamica(k-1, (1 << k) - 1));
    //debug_dinam();
    return 0;
}