Cod sursa(job #2546028)

Utilizator mjmilan11Mujdar Milan mjmilan11 Data 13 februarie 2020 19:25:02
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.6 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");

const int NMAX = 2005;
const int INF = (1<<29);

vector < pair<int,int> > v[NMAX];
queue <int> q;
int n,m,k;
int dp[20][131075],cost[20][2005];
int C[NMAX];

void bfs(int x)
{
    cost[x][C[x]]=0;
    q.push(C[x]);
    while(!q.empty())
    {
        int nod=q.front();
        q.pop();
        for(int i=0;i<v[nod].size();i++)
        {
            int vecin=v[nod][i].first;
            int lg=v[nod][i].second;
            if(cost[x][vecin]>cost[x][nod]+lg){
                cost[x][vecin]=cost[x][nod]+lg;
                q.push(vecin);
            }
        }
    }
}

int main()
{
    fin >> n >> m >> k;
    C[0]=1;
    int x,y,z;
    for(int i=1;i<=k;i++) fin >> C[i];
    C[k+1]=n;
    for(int i=1;i<=m;i++)
    {
        fin >> x >> y >> z;
        v[x].push_back({y,z});
        v[y].push_back({x,z});
    }
    for(int i=0;i<=k+1;i++)
    {
        for(int j=1;j<=n;j++) cost[i][j]=INF;
        for(int j=0;j<=(1<<(k+2));j++)
            dp[i][j]=INF;
    }
    for(int i=0;i<=k+1;i++) bfs(i);
    dp[0][1]=0;
    for(int config=0;config<(1<<k+2);config++)
    {
        for(int i=0;i<=k+1;i++)
        {
            if((config&(1<<i))!=0) continue;
            for(int j=0;j<=k+1;j++)
            {
                if((config&(1<<j))==0) continue;
                int nr=dp[j][config]+cost[j][C[i]];
                dp[i][config+(1<<i)]=min(dp[i][config+(1<<i)],nr);
            }
        }
    }
    fout << dp[k+1][(1<<(k+2))-1];
    return 0;
}