Cod sursa(job #2774219)

Utilizator stefanvoicaVoica Stefan stefanvoica Data 10 septembrie 2021 15:20:21
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.81 kb
#include <bits/stdc++.h>
#define dim 2002
using namespace std;
ifstream fin ("ubuntzei.in");
ofstream fout("ubuntzei.out");
int n,k,put[22],ord[dim],dp[18][131073],sol[dim];
const int inf=2000000000;

struct el
{
    int nod,cost,st;
    bool operator < (const el &A) const
    {
        return cost>A.cost;
    }
};
priority_queue<el> pq;
vector<el> a[dim],aa[dim];


void rezolva ()
{
    int i,stare;
    for (i=1; i<=k; i++)
        for (stare=1; stare<=put[k]-1; ++stare)
            dp[i][stare]=inf;
    dp[1][1]=0;
    for (stare=1;stare<put[k];stare++)
        for (i=1;i<=k;i++)
        if (dp[i][stare]!=inf)
        for (auto y:aa[i])
        dp[y.nod][stare|put[y.nod-1]]=min(dp[y.nod][stare|put[y.nod-1]],dp[i][stare]+y.cost);
    fout<<dp[k][put[k]-1];
}

void dijk ()
{
    while (!pq.empty())
    {
        el x=pq.top();
        pq.pop();
        if (x.cost==sol[x.nod])
            for (auto y: a[x.nod])
                if (x.cost+y.cost<sol[y.nod])
                {
                    sol[y.nod]=x.cost+y.cost;
                    pq.push({y.nod,sol[y.nod]});
                }
    }
}

int32_t main()
{
    int i,j,m,x,y,z;
    fin>>n>>m>>k;
    put[0]=1;
    for (i=1; i<=20; i++)
        put[i]=put[i-1]*2;
    for (i=1; i<=k; i++)
        fin>>ord[i];
    ord[++k]=1;
    ord[++k]=n;
    sort(ord+1,ord+k+1);
    for (i=1; i<=m; i++)
    {
        fin>>x>>y>>z;
        a[x].push_back({y,z,0});
        a[y].push_back({x,z,0});
    }
    for (i=1; i<=k; i++)
    {
        for (j=1; j<=n; j++)
            sol[j]=inf;
        sol[ord[i]]=0;
        pq.push ({ord[i],0});
        dijk();
        for (j=1; j<=k; j++)
            if (i!=j && sol[ord[j]]!=inf)
                aa[i].push_back({j,sol[ord[j]],0});
    }
    rezolva();
    return 0;
}