Cod sursa(job #2162968)

Utilizator MatteusTanase Matei Matteus Data 12 martie 2018 16:15:28
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.46 kb
#include <fstream>
#include <vector>
using namespace std;

ifstream cin ("dijkstra.in");
ofstream cout("dijkstra.out");

const int INF = 999999999;
const int N = 50005;

vector <int> a[N];
vector <int> c[N];

int x, y, cost, n, m, h[N], poz[N], d[N], p, q, nh;

void citire()
{
    int x,y,cost;
    cin>>n>>m;
    for(int i=0; i<=m; i++)
    {
        cin>>x>>y>>cost;
        a[x].push_back(y);
        c[x].push_back(cost);
    }
}

void swapp(int p, int q)
{
    swap(h[p], h[q]);
    poz[h[p]]=p;
    poz[h[q]]=q;
}

void urca(int p)
{
    while(p>1 && d[h[p]]<d[h[p/2]])
    {
        swapp(p, p/2);
        p/=2;
    }
}

void coboara(int p)
{
    int fs=2*p, fd=2*p+1, bun=p;
    if(fs<=nh && d[h[fs]]<d[h[bun]])
    {
        bun=fs;
    }
    if(fd<=nh && d[h[fd]]<d[h[bun]])
    {
        bun=fd;
    }
    if(bun!=p)
    {
        swapp(p, bun);
        coboara(bun);
    }
}

void adauga(int p)
{
    h[++nh]=p;
    poz[p]=nh;
    urca(nh);
}

void sterge(int p)
{
    swapp(p, nh--);
    urca(p);
    coboara(p);
}

void dijkstra(int x0)
{
	nh=n;
	for(int i=1; i<=n; i++)
	{
		d[i]=INF;
		h[i]=i;
		poz[i]=i;
	}
	d[x0]=0;
	urca(poz[x0]);
	while(nh>0 && d[h[1]] != INF)
	{
		int x=h[1];
		sterge(1);
		for(int i=0; i<a[x].size(); i++)
		{
			int y=a[x][i];
			cost=c[x][i];
			if(d[x]+cost<d[y])
			{
				d[y]=d[x]+cost;
				//pred[y]=x;
				urca(poz[y]);
			}
		}
	}
}


int main()
{
    citire();
    dijkstra(1);
    for(int i=2; i<=n; i++)
        cout<<d[i]<<' ';
    return 0;
}