Cod sursa(job #967994)

Utilizator vali.lazarLazar Valentin vali.lazar Data 28 iunie 2013 23:48:36
Problema Ubuntzei Scor 85
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.18 kb
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

#define Nmax 2010
#define PI pair<int, int>
#define pb push_back
#define mp make_pair
#define f first
#define s second
#define Inf 0x3f3f3f3f

int N, M, X, Y, Z, K, C[16], Dist[Nmax][Nmax], Dp[1 << 16][16];
bool InQueue[Nmax];
vector<PI> G[Nmax];

void BellmanFord(int StartNode, int V[])
{
    for(int i = 1; i <= N; ++ i)
    {
        InQueue[i] = 0;
        V[i] = Inf;
    }
    queue<int> Q;
    Q.push(StartNode);
    InQueue[StartNode] = 1;
    V[StartNode] = 0;
    while(!Q.empty())
    {
        int Node = Q.front();
        Q.pop();
        InQueue[Node] = 0;
        for(vector<PI> :: iterator it = G[Node].begin(); it != G[Node].end(); ++ it)
            if(V[it -> f] > V[Node] + it -> s)
            {
                V[it -> f] = V[Node] + it -> s;
                if(!InQueue[it -> f])
                {
                    InQueue[it -> f] = 1;
                    Q.push(it -> f);
                }
            }
    }
}

int main()
{
    freopen("ubuntzei.in", "r", stdin);
    freopen("ubuntzei.out", "w", stdout);

    scanf("%i %i", &N, &M);
    scanf("%i", &K);
    for(int i = 0; i < K; ++ i)
        scanf("%i", &C[i]);
    for(int i = 1; i <= M; ++ i)
    {
        scanf("%i %i %i", &X, &Y, &Z);
        G[X].pb(mp(Y, Z));
        G[Y].pb(mp(X, Z));
    }

    for(int i = 1; i <= N; ++ i)
        BellmanFord(i, Dist[i]);

    if(K == 0)
    {
        printf("%i\n", Dist[1][N]);
        return 0;
    }
    memset(Dp, Inf, sizeof(Dp));

    for(int i = 0; i < K; ++ i) Dp[1 << i][i] = Dist[1][ C[i] ];

    for(int Conf = 1; Conf < (1 << K); ++ Conf)
        for(int i = 0; i < K; ++ i)
            if((Conf & (1 << i)))
                for(int j = 0; j < K; ++ j)
                    if(i != j && Conf & (1 << j))
                        Dp[Conf][i] = min(Dp[Conf][i], Dp[Conf ^ (1 << i)][j] + Dist[ C[j] ][ C[i] ]);

    int Ans = Inf;
    for(int i = 0; i < K; ++ i)
            Ans = min(Ans, Dp[(1 << K) - 1][i] + Dist[ C[i] ][N]);

    printf("%i\n", Ans);
    return 0;
}