Cod sursa(job #2704855)

Utilizator darisavuSavu Daria darisavu Data 11 februarie 2021 15:05:57
Problema Ubuntzei Scor 45
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.79 kb
#include <bits/stdc++.h>
#define inf 100000000
using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");
int a[16],k,n,m,x,y,c,d[10005],viz[10005],fr[20],sol[20],mn=inf;
int p[2005][2005];
priority_queue< pair<int,int>, vector<pair<int,int> >, greater<pair<int,int> > >q;
vector<pair<int,int> >v[10005];
void dijkstra(int x)
{
    int nod,sum=0;
    for(int i=1; i<=n; i++)
    {
        d[i]=inf;
        viz[i]=0;
    }
    q.push({0,x});
    d[x]=0;
    while(!q.empty())
    {
        nod=q.top().second;
        q.pop();
        if(viz[nod]==0)
        {
            for(int i=0; i<v[nod].size(); i++)
            {
                if(d[nod]+v[nod][i].first<d[v[nod][i].second])
                {
                    d[v[nod][i].second]=d[nod]+v[nod][i].first;
                    q.push({d[v[nod][i].second],v[nod][i].second});
                }
            }

        }
        viz[nod]=1;
    }
    for(int i=1; i<=n; i++)
    {
        p[x][i]=d[i];
    }
}
void verif(int nod)
{
    int sum=0;
    for(int i=1; i<=k; i++)
    {
        sum+=p[nod][a[sol[i]]];
        nod=a[sol[i]];
    }
    sum+=p[nod][n];
    if(sum<mn) mn=sum;
}
void bkt(int p)
{
    if(p>k)
    {
        verif(1);
    }
    for(int i=1; i<=k; i++)
    {
        if(fr[i]==0)
        {
            fr[i]=1;
            sol[p]=i;
            bkt(p+1);
            fr[i]=0;

        }
    }
}
int main()
{
    int i,j;
    f>>n>>m;
    f>>k;
    for(i=1; i<=k; i++)
        f>>a[i];
    for(i=1; i<=m; i++)
    {
        f>>x>>y>>c;
        v[x].push_back({c,y});
        v[y].push_back({c,x});
    }
    dijkstra(1);
    for(i=1; i<=k; i++)
    {
        dijkstra(a[i]);
    }
    dijkstra(n);
    bkt(1);
    g<<mn<<" ";
    return 0;
}