Cod sursa(job #2686040)

Utilizator MateGMGozner Mate MateGM Data 18 decembrie 2020 14:00:22
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.4 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <list>

using namespace std;

const int inf=1<<30;

int dijkstra(int n,int k,const vector<vector<pair<int,int> > >&adj,vector<int> &h)
{
    vector<vector<int> >dist(n+1,vector<int>((1<<k),inf));
    vector<vector<bool> >visit(n+1,vector<bool>((1<<k),false));
    //priority_queue<pair<int,pair<int,int> > >q;
    priority_queue<pair<int,pair<int,int> >,vector<pair<int,pair<int,int> > >,greater<pair<int,pair<int,int> > > >q;
    q.push({0,{1,0}});
    dist[1][0]=0;
    while(!q.empty())
    {
        auto cd=q.top().first;
        auto cn=q.top().second.first;
        auto cs=q.top().second.second;
        q.pop();
        visit[cn][cs]=true;


        if(cd==dist[cn][cs])
        {
            if((h[cn]) &&((cs&h[cn])==0)){
                auto newset=cs|h[cn];
                if((!visit[cn][newset]) && (dist[cn][newset]>dist[cn][cs]))
                {
                    dist[cn][newset]=dist[cn][cs];
                    q.push({dist[cn][newset],{cn,newset}});
                }
            }

        }

        for(auto p :adj[cn])
        {
            int v=p.first;
            int e=p.second;
            if((!visit[v][cs]) && dist[v][cs]>e+dist[cn][cs])
            {
                dist[v][cs]=e+dist[cn][cs];
                q.push({dist[v][cs],{v,cs}});
            }
        }
    }
    return dist[n][(1<<k)-1];
}

void dijkstra1(int idx,vector<vector<pair<int,int> > >&adj,vector<vector<int> >&dist)
{
    dist[idx][idx]=0;
    priority_queue<pair<int,int >,vector<pair<int,int > >,greater<pair<int,int> > >q;
    q.push({0,idx});
    while(!q.empty())
    {
        auto best=q.top();
        int node=best.second;
        q.pop();
       // if(cost>dist[start][node])continue;

        for(auto p:adj[node])
        {
            int v=p.first;
            int e=p.second;
            if(  dist[idx][v]>dist[idx][node]+e)
            {
                dist[idx][v]=dist[idx][node]+e;
                dist[v][idx]=dist[idx][v];
                q.push({dist[idx][v],v});
            }
        }

    }


}

int main()
{
    ifstream be("ubuntzei.in");
    ofstream ki("ubuntzei.out");

    int n,m,k;
    be>>n>>m>>k;
    vector<int>h(20);
    for(int i=1;i<=k;i++)
    {
        int f;
        be>>f;
        h[i]=f;
    }
    vector<vector<pair<int,int> > >adj(n+1);
    for(int i=1;i<=m;i++)
    {
        int x,y,cost;
        be>>x>>y>>cost;
        adj[x].push_back({y,cost});
        adj[y].push_back({x,cost});
    }
    vector<vector<int> >dist(2005,vector<int>(2005));
    h[0]=1,h[++k]=n;
    for(int i=1;i<=k;i++)
        for(int j=1;j<=n;j++)
            dist[h[i]][j]=inf;

    for(int i=0;i<=k;i++){
        dijkstra1(h[i],adj,dist);
    }
    vector<vector<int> >d(k+3,vector<int>((1<<k)+3));

    for (int i=0; i<=k; i++)
        for (int j=1; j<=(1<<k); j++)
            d[i][j] = inf;
    d[0][1]=0;



    for(int mask=1;mask<(1<<k);mask++)
        for(int i=0;i<k;i++)
            if(mask&(1<<i))
                for(int j=0;j<k;j++){
                    if(i!=j && (mask&(1<<j)))
                        d[i][mask]=min(d[i][mask],d[j][mask-(1<<i)]+dist[h[j]][h[i]]);
                }
    int sol=inf;
    for(int i=0;i<=k;i++)
        sol=min(sol,d[i][(1<<k)-1]+dist[h[i]][n]);
    ki<<sol<<endl;
    return 0;
}