Cod sursa(job #573859)

Utilizator mihai995mihai995 mihai995 Data 6 aprilie 2011 17:00:14
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <fstream>
using namespace std;

const int N=50001,M=250001,inf=1<<30;
struct arc{int x,y,c;} a[M];
int v[M],dist[N],s[N],n,m;
bool use[N];

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

inline bool comp(arc a,arc b)
{
	return a.x<b.x;
}

inline void sch(int &a,int &b)
{
	int c=a;a=b;b=c;
}

inline bool cmp(int a,int b)
{
	return dist[a]<dist[b];
}

inline void up(int x)
{
	if (x>1 && cmp(v[x],v[x>>1]))
	{
		sch(v[x],v[x>>1]);
		up(x>>1);
	}
}

inline void down(int x)
{
	int m=x;
	if ((x<<1)<=n && cmp(v[x<<1],v[x]))
		m=(x<<1);
	if ((x<<1)<n && cmp(v[(x<<1)+1],v[x]))
		m=(x<<1)+1;
	if (m!=x)
	{
		sch(v[x],v[m]);
		down(m);
	}
}

inline int del()
{
	int q=v[1];
	v[1]=v[v[0]--];
	down(1);
	return q;
}

inline void add(int x)
{
	v[++v[0]]=x;
	up(v[0]);
}

void dijkstra(int x)
{
	int i,y,c;
	dist[x]=0;
	add(x);
	while (v[0])
	{
		x=del();
		if (use[x])
			continue;
		use[x]=true;
		for (i=s[x];i<s[x+1];i++)
		{
			y=a[i].y;c=a[i].c;
			if (dist[y]>dist[x]+c)
			{
				dist[y]=dist[x]+c;
				add(y);
			}
		}
	}
}

int main()
{
	int i,j;
	in>>n>>m;
	for (i=1;i<=n;i++)
	{
		in>>a[i].x>>a[i].y>>a[i].c;
		dist[i]=inf;
	}
	sort(a+1,a+n+1,comp);
	for (i=j=1,s[n+1]=m+1;i<=n;s[i]=j,i++)
		for (;a[j].x<i && j<=m;j++);
	dijkstra(1);
	for (i=2;i<=n;i++)
		out<<dist[i]<<" ";
	out<<"\n";
	return 0;
}