Cod sursa(job #2185658)

Utilizator AndreiD31Dragan Andrei AndreiD31 Data 24 martie 2018 18:55:53
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.38 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)
    {
        fv[D[p].poz]=p/2;
        fv[D[p/2].poz]=p;
        swap(D[p],D[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)
        {
            fv[D[p].poz]=2*p;
            fv[D[2*p].poz]=p;
            swap(D[2*p],D[p]);
            down(2*p);
        }
        else
        {
            fv[D[p].poz]=2*p+1;
            fv[D[2*p+1].poz]=p;
            swap(D[2*p+1],D[p]);
            down(2*p+1);
        }
    }
    else if(2*p<=n&&D[p].va>D[2*p].va)
    {
        fv[D[p].poz]=2*p;
        fv[D[2*p].poz]=p;

        swap(D[p],D[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;
    }
    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]=min(D[1].poz,mini);
        if(mini!=INT_MAX/2)
        {
            D[1].va=INT_MAX/2;
            for(i=0; i<a[x].size(); i++)
            {
                if(mini+d[x][i]<D[fv[a[x][i]]].va)
                {
                    D[fv[a[x][i]]].va=mini+d[x][i];
                    up(fv[a[x][i]]);
                }
            }
        down(1);
        }

    }
}



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(1);

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