Cod sursa(job #2720076)

Utilizator maraboneaMara Bonea marabonea Data 10 martie 2021 16:10:43
Problema Ubuntzei Scor 25
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.91 kb
/**
*/
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
const int nmax=10000;
const int N=2020;
const int oo = 2000000000;
vector< pair <int,int> > v[nmax];
int n,m,k,c[16],s[N+5],mini,st[200];
int minim=oo;
void read()
{
    int x,y,z;
    fin>>n>>m;
    fin>>k;
    for(int i=1;i<=k;i++)
    {
        fin>>c[i];
    }
    for(int i=1; i<=m; i++)
    {
        fin>>x>>y>>z;
        v[x].push_back(make_pair(y,z));
        v[y].push_back(make_pair(x,z));
    }
}
int d[N][N];
void dijkstra(int x)
{
    for(int i=1; i<=n; i++)
    {
        d[x][i]=oo;
        s[i]=0;
    }
    d[x][x]=0;
    for(int k=1; k<n; k++)
    {
        int minim=oo,nod;
        for(int i=1; i<=n; i++)
        {
            if(d[x][i]<minim && s[i]==0)
            {
                minim=d[x][i];
                nod=i;
            }
        }
        s[nod]=1;
        for(int i=0; i<v[nod].size(); i++)
        {
            int vecin=v[nod][i].first, cost=v[nod][i].second;
            d[x][vecin]=min(d[x][vecin],d[x][nod]+cost);
            d[vecin][x]=d[x][vecin];
        }
    }
}
void afisare()
{
    mini=d[1][st[1]];
    for(int i=1;i<k;i++)
    {
        mini=mini+d[st[i]][st[i+1]];
    }
    mini=mini+d[st[k]][n];

    minim=min(minim,mini);
}
int valid1(int x)
{
    for(int i=1;i<x;i++)
    {
        if(st[i]==st[x])
            return 0;
    }
    return 1;
}
void back(int x)
{
    if(x>k)
        afisare();
    else
    {
        for(int i=1;i<=k;i++)
        {
            st[x]=c[i];
            if(valid1(x))
                back(x+1);
        }
    }
}
void solve()
{

    c[0]=1;
    c[k+1]=n;
    for(int i=0; i<=k+1; i++)
    {
        dijkstra(c[i]);
    }
    back(1);
    fout<<minim;
}
int main()
{
    read();
    solve();
    return 0;
}