Cod sursa(job #1121675)

Utilizator cristitamasTamas Cristian cristitamas Data 25 februarie 2014 13:38:43
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.98 kb
#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#define INF 0x3f3f3f3f
#define Nmax 2050
#define Minim(x,y) (x<y?x:y)
using namespace std;

vector <pair <int, int> > Graf[Nmax];
priority_queue < pair <int, int>, vector <pair <int, int> >, greater <pair <int, int> > > Coada;
int D[Nmax][Nmax];

int N,M,K;
int Ksir[20];

int DP[20][1<<20];
int Minim;


void Citire()
{
    int x,y,c;
    scanf("%d %d",&N,&M);
    scanf("%d ",&K);
    Ksir[1]=1;
    for(int i=2;i<=K+1;++i)
            scanf("%d ",&Ksir[i]);
    K++;
    Ksir[++K]=N;
    for(int i=1;i<=M;++i)
    {
        scanf("%d %d %d",&x,&y,&c);
        Graf[x].push_back(make_pair(c,y));
        Graf[y].push_back(make_pair(c,x));
    }
    for(int i=1;i<=N;++i)
        for(int j=1;j<=N;++j)
            if(i!=j)
                D[i][j]=INF;

}

void Dijkstra(int i)
{
    Coada.push(make_pair(0,i));
    while(!Coada.empty())
    {
        int t=Coada.top().second;
        Coada.pop();
        for(int j=0;j<Graf[t].size();++j)
        {
            int v=Graf[t][j].second;
            int c=Graf[t][j].first;
            if(D[i][v]>D[i][t]+c)
            {
                D[i][v]=D[i][t]+c;
                Coada.push(make_pair(D[i][v],v));
            }
        }
    }
}


void Formare()
{
    for(int s=1;s<(1<<K);++s)
        for(int i=1;i<=K;++i)
            DP[i][s]=INF;
    for(int i=1;i<=K;++i)
        DP[i][(1<<i)+1]=D[1][i];
}

int Rezolvare(int i,int s)
{
    if(DP[i][s]!=INF)
        return DP[i][s];
    if(s%2==0)
        return DP[i][s]=INF-1;
    int Min=DP[i][s];
    for(int j=1;j<=K;++j)
        Min=Minim(Min,Rezolvare(j,s-1<<i))+D[j][i];
    DP[i][s]=Min;
    return DP[i][s];
}


int main()
{
    freopen("ubuntzei.in","r",stdin);
    freopen("ubuntzei.out","w",stdout);
    Citire();
    for(int i=1;i<=K+2;++i)
        Dijkstra(Ksir[i]);
    Formare();
    printf("%d", Rezolvare(K,(1<<K)-1));
    return 0;
}