Cod sursa(job #2417921)

Utilizator AlexandruBrezuleanuAlexandruBrezuleanu AlexandruBrezuleanu Data 2 mai 2019 12:32:48
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.12 kb
#include <fstream>
#include <vector>
#define MAX 50010
#define INF 10000000
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
struct nod
{
    int y,c;
};
vector <nod>g[MAX];
struct heap
{
    int val,pos;
}
h[MAX];
int n,m,lgh;
int d[MAX];
bool uz[MAX];
void insertheap(int &lgh,int x,int pozi);
int extractheap(int &lgh);
void citire();
void dijkstra(int node);
void afisare();
int main()
{
    citire();
    dijkstra(1);
    afisare();
    return 0;
}
void citire()
{
    int i,x,y,cost;
    fin>>n>>m;
    for(i=1; i<=m; i++)
    {
        fin>>x>>y>>cost;
        g[x].push_back({y,cost});
    }
    for(i=2; i<=n; i++)
        d[i]=INF;
}
void dijkstra(int node)
{
    int i,vecin,cost,nodecrt;
    insertheap(lgh,d[node],node);
    uz[node]=1;
    while(lgh)
    {
        nodecrt=extractheap(lgh);
        uz[nodecrt]=0;
        for(i=0; i<g[nodecrt].size(); i++)
        {
            vecin=g[nodecrt][i].y;
            cost=g[nodecrt][i].c;
            if(d[vecin]>d[nodecrt]+cost)
            {
                d[vecin]=d[nodecrt]+cost;
                if(!uz[vecin])
                {
                    insertheap(lgh,d[vecin],vecin);
                    uz[vecin]=1;
                }
            }
        }
    }
}
void afisare()
{
    int i;
    for(i=2; i<=n; i++)
        fout<<d[i]<<' ';
}
void insertheap(int &lgh,int x,int pozi)
{
    int tata,fiu,aux;
    lgh++;
    fiu=lgh;
    tata=lgh/2;
    h[fiu].val=x;
    h[fiu].pos=pozi;
    while(tata>0&&h[tata].val>h[fiu].val)
    {
        swap(h[tata],h[fiu]);
        fiu=tata;
        tata=fiu/2;
    }
}
int extractheap(int &lgh)
{
    int tata,fiu,x;
    tata=1;
    fiu=2;
    x=h[1].pos;
    h[1].val=h[lgh].val;
    h[1].pos=h[lgh].pos;
    lgh--;
    while(fiu<=lgh)
    {
        if(fiu+1<=lgh)
            if(h[fiu].val>h[fiu+1].val)
                fiu++;
        if(h[tata].val>h[fiu].val)
        {
            swap(h[tata],h[fiu]);
            tata=fiu;
            fiu*=2;
        }
        else
            break;
    }
    return x;
}