Cod sursa(job #2185692)

Utilizator AndreiD31Dragan Andrei AndreiD31 Data 24 martie 2018 19:21:28
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.45 kb
#include <fstream>
#include <climits>
#include <vector>
using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

vector <int> a[50010];
vector <int> d[50010];

int n,fv[50100],w[50010],viz[50010];

struct dist
{
    int poz,va;
} D[50010];


void up(int p)
{
    if(p>1&&D[p].va<D[p/2].va)
    {

        swap(D[p],D[p/2]);
        fv[D[p].poz]=p;
        fv[D[p/2].poz]=p/2;

        up(p/2);
    }
}


void down(int p)
{
    if(2*p+1<=n&&(D[p].va>D[2*p].va||D[p].va>D[2*p+1].va))
    {
        if(D[2*p].va<D[2*p+1].va)
        {

            swap(D[2*p],D[p]);
            fv[D[p].poz]=p;
            fv[D[2*p].poz]=2*p;
            down(2*p);
        }
        else
        {

            swap(D[2*p+1],D[p]);
            fv[D[p].poz]=p;
            fv[D[2*p+1].poz]=2*p+1;
            down(2*p+1);
        }
    }
    else if(2*p<=n&&D[p].va>D[2*p].va)
    {
        swap(D[p],D[2*p]);
         fv[D[p].poz]=p;
        fv[D[2*p].poz]=2*p;
    }
}






void dijkstra (int s)
{
    int i,mini,x,k;
    for(i=1; i<=n; i++)
    {
        D[i].va=INT_MAX/2;
        D[i].poz=i;
    }
    w[s]=0;
    D[s].va=0;
    up(s);


    for(k=1; k<=n; k++)
    {

        /* mini=INT_MAX;
         x=0;
         for(i=1; i<=n; i++)
             if(viz[i]==0&&D[i]<mini)
             {
                 mini=D[i];
                 x=i;
             }
        */
        mini=D[1].va;
        x=D[1].poz;


      //  for(i=1;i<=n;i++)
      //      g<<D[i].va<<" "<<D[i].poz<<'\n';
     //   g<<'\n';

       // w[D[1].poz]=D[1].poz;
        if(mini!=INT_MAX/2)
        {
              D[1].va=INT_MAX/2;
        down(1);
            for(i=0; i<a[x].size(); i++)
            {
                if(mini+d[x][i]<w[a[x][i]])
                {

                      w[a[x][i]]=mini+d[x][i];
                       D[fv[a[x][i]]].va=mini+d[x][i];
                      up(fv[a[x][i]]);
                }
            }


        }

    }
}



int m,x,y,val,i;
int main()
{
    f>>n>>m;

    for(i=1; i<=n; i++)
    {
        D[i].va=INT_MAX/2;
        D[i].poz=i;
        fv[i]=i;
        w[i]=INT_MAX/2;
    }


    for(i=1; i<=m; i++)
    {
        f>>x>>y>>val;
        a[x].push_back(y);
        d[x].push_back(val);
    }

    dijkstra(2);

    for(i=2; i<=n; i++)
        if(w[i]==INT_MAX/2)g<<0<<" ";
        else g<<w[i]<<" ";
    return 0;
}