Cod sursa(job #1905717)

Utilizator mateigabriel99Matei Gabriel mateigabriel99 Data 6 martie 2017 10:28:00
Problema Ubuntzei Scor 35
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <bits/stdc++.h>

#define NMax 2005

using namespace std;

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

int N,M,K,C[NMax];
int cost[NMax][NMax];

void Roy_Floyd()
{
    for(int k=1;k<=N;k++)
        for(int i=1;i<=N;i++)
            for(int j=1;j<=N;j++)
                if(cost[i][k] && cost[k][j] && i!=j && ( cost[i][j]>cost[i][k]+cost[k][j] || !cost[i][j]))
                    cost[i][j]=cost[i][k]+cost[k][j];
}

int main()
{
    fin>>N>>M;
    fin>>K;
    for(int i=1;i<=K;i++)
        fin>>C[i];

    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++)
            if(i!=j)
                cost[i][j]=1<<20;

    for(int i=1;i<=M;i++)
    {
        int x,y,z;
        fin>>x>>y>>z;
        cost[x][y]=cost[y][x]=z;
    }

    Roy_Floyd();

    if(K==0)
    {
        fout<<cost[1][N];
        return 0;
    }

    int minDist=1<<30;
    do{
        int dist=0;
        dist+=cost[1][C[1]];
        for(int i=1;i<K;i++)
            dist+=cost[C[i]][C[i+1]];
        dist+=cost[C[K]][N];

        minDist=min(minDist,dist);
    }while(next_permutation(C+1,C+K+1));
    fout<<minDist;

    return 0;
}