Cod sursa(job #1914251)

Utilizator CriistinaMicula Cristina Criistina Data 8 martie 2017 16:06:32
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.04 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
#include <queue>
#define Dim 1<<15
#define Kmax 15
#define Nmax 2001
#define Max 1000000000

using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");

int dp[Dim][Kmax], sol=Max;
short n, m, k, ob[Kmax];
vector <pair<int, int>> gr[Nmax];


int cost(int x, int y)
{
    int rez=Max;
    queue<pair<int, int> > c;
    int viz[Nmax];
    for(int i=0;i<=n;i++)
        viz[i]=Max;
    c.push(make_pair(x,0));
    viz[x]=0;
    while(!c.empty())
    {
        pair<int, int> fr=c.front();
        c.pop();
        for(auto i:gr[fr.first])
        {
            if(i.first==y)
                rez=min(rez, fr.second+i.second);
            if(viz[i.first]>fr.second+i.second)
            {
                viz[i.first]=fr.second+i.second;
                c.push(make_pair(i.first, fr.second+i.second));
            }
        }
    }
    if(rez==Max)
        return -1;
    return rez;
}
int main()
{
    f>>n>>m;
    f>>k;
    for(int i=0;i<k;i++)
        f>>ob[i];
    for(int i=1;i<=m;i++)
    {
        int x,y,c;
        f>>x>>y>>c;
        gr[x].push_back(make_pair(y,c));
        gr[y].push_back(make_pair(x,c));
    }
    if(k==0)
    {
        g<<cost(1,n);
        return 0;
    }
    for(int i=0;i<(1<<k);i++)
        for(int j=0;j<k;j++)
            dp[i][ob[j]]=Max;
    for(int i=0;i<k;i++)
        dp[1<<i][ob[i]]=cost(1, ob[i]);
    for(int nr=0;nr<k;nr++)
    {
        for(int conf=0;conf<(1<<k);conf++)
        {
            if((conf&(1<<nr))!=0)
            {
                for(int i=0;i<k;i++)
                {
                    if(i!=nr && (conf&(1<<i))!=0)
                    {
                        dp[conf][ob[nr]]=min(dp[conf][ob[nr]], dp[conf^(1<<nr)][ob[i]]+cost(ob[i], ob[nr]));
                    }
                }
            }
        }
    }
    for(int i=0;i<k;i++)
    {
        sol=min(sol, dp[(1<<k)-1][ob[i]]+cost(ob[i], n));
    }
    g<<sol;
    return 0;
}