Cod sursa(job #2685501)

Utilizator MateGMGozner Mate MateGM Data 17 decembrie 2020 09:45:21
Problema Ubuntzei Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.93 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];
}

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

    int n,m,k;
    be>>n>>m>>k;
    vector<int>h(n+1,0);
    for(int i=0;i<k;i++)
    {
        int f;
        be>>f;
        h[f]=1<<i;

    }
    vector<vector<pair<int,int> > >adj(n+1);
    for(int i=0;i<m;i++)
    {
        int x,y,cost;
        be>>x>>y>>cost;
        adj[x].push_back({y,cost});
        adj[y].push_back({x,cost});
    }

    ki<<dijkstra(n,k,adj,h);
    return 0;
}