Cod sursa(job #2864567)

Utilizator MokaDomos Mozes Moka Data 7 martie 2022 23:00:32
Problema Ubuntzei Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.36 kb
#include <list>
#include <fstream>
#include <queue>
#include <vector>
struct szomszed
{
    int vegpont;
    int suly;
};
struct onmaga
{
    int szam;
    unsigned long long osszeg=0;
};
struct CompareSuly
{
    bool operator()(onmaga &p1, onmaga &p2)
    {
        return p1.osszeg > p2.osszeg;
    }
};
using namespace std;
ifstream cin("ubuntzei.in");
ofstream cout("ubuntzei.out");
int m,n,kdb;
list<szomszed> v[2001];
int varos[20];
bool k[2001];
priority_queue<onmaga,vector<onmaga>,CompareSuly> pq;
unsigned long long minm=0;
bool jart[2001];
unsigned long long tav[2001];
onmaga aktualis;
onmaga on;
void dijsktra(int kezd,int veg)
{
    while(!pq.empty())
        pq.pop();
    int belson=veg;
    on.szam=kezd;
    on.osszeg=0;
    pq.push(on);
    for (int i=0; i<=2000; i++)
    {
        tav[i]=184467440737095516;
    }
    for (int i=0; i<=2000; i++)
    {
        jart[i]=false;
    }
    tav[kezd]=0;
    bool megvan=true;
    while (!pq.empty() && megvan)
    {
        aktualis = pq.top();
        if(aktualis.szam==belson)
        {
            minm+=aktualis.osszeg;
            //cout<<aktualis.osszeg<<endl;
            megvan=false;
        }
        pq.pop();
        if(megvan)
        {
            jart[aktualis.szam]=true;
            for (szomszed l : v[aktualis.szam])
            {
                unsigned long long tavolsag=aktualis.osszeg+l.suly;
                if(jart[l.vegpont]!=true)
                {
                    if(tav[l.vegpont]>tavolsag)
                    {
                        tav[l.vegpont]=tavolsag;
                        onmaga ledik;
                        ledik.szam=l.vegpont;
                        ledik.osszeg=tav[l.vegpont];
                        pq.push(ledik);
                    }
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m>>kdb;
    varos[0]=1;
    for (int i=1; i<=kdb; i++)
    {
        cin>>varos[i];
    }
    varos[kdb+1]=n;
    for (int i=0; i<m; i++)
    {
        int a;
        int b;
        int c;
        cin>>a>>b>>c;
        szomszed l;
        l.vegpont=b;
        l.suly=c;
        v[a].push_back(l);
        l.vegpont=a;
        v[b].push_back(l);
    }
    for (int i=0; i<=kdb+1; i++)
    {
        dijsktra(varos[i],varos[i+1]);
    }
    cout<<minm;
    return 0;
}