Cod sursa(job #2009747)

Utilizator tanasaradutanasaradu tanasaradu Data 10 august 2017 19:16:19
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.23 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
const int kmax=16;
const int bitimax=(1<<kmax);
const int nmax=2001;
const int oo=(1<<30);
vector<pair<int,int> >L[nmax];
int matd[nmax][nmax],dp[bitimax][kmax],prieteni[kmax],n,m,dist[nmax],valmax,k;
priority_queue<pair<int,int> >mn;
bitset<nmax>viz;
inline void Read()
{
    fin>>n>>m;
    fin>>k;
    for(int i=0;i<k;i++)
        fin>>prieteni[i];
    for(int i=1;i<=m;i++)
    {
        int x,y,cost;
        fin>>x>>y>>cost;
        L[x].push_back({y,cost});
        L[y].push_back({x,cost});

    }
    valmax=(1<<k)-1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            matd[i][j]=oo;
}
inline void Dijkstra(int varf)
{
    for(int i=1;i<=n;i++)
        dist[i]=oo;
    dist[varf]=0;
    mn.push({dist[varf],varf});
    viz.reset();
    while(!mn.empty())
    {
        int x=mn.top().second;
        mn.pop();
        if(!viz[x])
        {
            viz[x]=1;
            for(int i=0;i<L[x].size();i++)
            {
                int p=L[x][i].first;
                int q=L[x][i].second;
                if(dist[p]>dist[x]+q)
                {
                    dist[p]=dist[x]+q;
                    mn.push({-dist[p],p});
                }
            }
        }
    }
    for(int i=1;i<=n;i++)
        matd[varf][i]=min( matd[varf][i],dist[i]);
}
inline void Build()
{
    for(int i=0;i<=valmax;i++)
        for(int j=0;j<k;j++)
            dp[i][j]=oo;
    Dijkstra(1);
    for(int i=1;i<=k;i++)
        Dijkstra(prieteni[i]);
    Dijkstra(n);
}
inline void Solve()
{
    for(int i=0;i<k;i++)
        dp[1<<i][i]=matd[1][prieteni[i]];
    for(int i=0;i<=valmax;i++)
        for(int j=0;j<k;j++)
            if((i&(1<<j)))
        for(int pas=0;pas<k;pas++)
            if(!(i&(1<<pas)))
                dp[i|(1<<pas)][pas]=min(dp[i|(1<<pas)][pas],dp[i][j]+matd[j][pas]);

    int sol=oo;
    for(int i=0;i<k;i++)
        sol=min(sol,dp[valmax][i]+matd[n][prieteni[i]]);
    if(k==0)
        fout<<matd[1][n]<<"\n";
    else fout<<sol<<"\n";
}
int main()
{
    Read();
    Build();
    Solve();
    fin.close();
    fout.close();
    return 0;
}