Cod sursa(job #2322345)

Utilizator vladsirbu23Vlad Sirbu vladsirbu23 Data 17 ianuarie 2019 18:23:29
Problema Ubuntzei Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.41 kb
#include <iostream>
#include <fstream>
#define pinf 10000000000010
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
long long N,M,a[2010][2010],i,j,k,o[20],x,y,z,m[32770][17],c,rez,K,l;
int main()
{

    fin>>N>>M>>K;
    for(i=1; i<=N; i++)
        for(j=1; j<=N; j++)
            if(i!=j)
                a[i][j]=pinf;
    for(i=0; i<K; i++)
        fin>>o[i];
    for(i=1; i<=M; i++)
    {
        fin>>x>>y>>z;
        a[x][y]=z;
        a[y][x]=z;
    }
    for(k=1; k<=N; k++)
    {
        for(i=1; i<=N; i++)
        {
            for(j=1; j<=N; j++)
            {
                if(a[i][j]>a[i][k]+a[k][j])
                {
                    a[i][j]=a[i][k]+a[k][j];
                }
            }
        }
    }
    for(i=1; i<(1<<K); i++)
    {
        for(j=0; j<K; j++)
            m[i][j]=pinf;
    }
    c=-1;

    for(i=1; i<(1<<K); i=i<<1)
    {
        c++;
        m[i][c]=a[o[c]][1];
    }
    for(i=1; i<(1<<K); i++)
        for(l=0; l<K; l++)
        {
            if((i&(1<<l))==(1<<l))
                for(j=0; j<K; j++)
                    if((i&(1<<j))!=(1<<j))
                    {
                        m[i+(1<<j)][j]=min(m[i+(1<<j)][j],m[i][l]+a[o[l]][o[j]]);

                    }
        }
   i=(1<<K)-1;
   rez=pinf;
   for(j=0;j<K;j++)
   {
      rez=min(rez,m[i][j]+a[N][o[j]]);
   }
   fout<<rez;
}