Cod sursa(job #2347979)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 19 februarie 2019 11:45:35
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define I 1<<30
#define F first
#define S second
#define N 50001
#define M 3000000
set<pair<int,int>> c;
int n,m,d[N],x,y,z,b,i,j,k,o,l;
vector<pair<int,int>> a[N];
char r[M];
int A()
{
  	int s=0;
  	for(;r[o]<'0'||r[o]>'9';o++);
  	for(;r[o]>='0'&&r[o]<='9';o++)
  		s=s*10+r[o]-'0';
  	return s;
}
void S(int b)
{
    char e[100];
    int j;
    if(!b)
        r[l++]=48;
    for(j=0;b;b/=10,j++)
        e[j]=b%10+48;
    for(j--;j>=0;j--)
        r[l++]=e[j];
}
int main()
{
    freopen("dijkstra.in","r",stdin),freopen("dijkstra.out","w",stdout),fread(r,1,M,stdin),n=A(),m=A();
	while(m--)
	  	x=A(),y=A(),z=A(),a[x].pb({y,z});
    for(i=2;i<=n;i++)
        d[i]=I;
	for(c.insert({0,1});!c.empty();)
		for(i=c.begin()->S,c.erase(c.begin()),k=a[i].size(),j=0;j<k;j++)
            if(d[i]+a[i][j].S<d[a[i][j].F])
            {
                if(d[a[i][j].F]!=I)
                    c.erase(c.find({d[a[i][j].F],a[i][j].F}));
                d[a[i][j].F]=d[i]+a[i][j].S,c.insert({d[a[i][j].F],a[i][j].F});
            }
	for(i=2;i<=n;i++)
        S((d[i]==I?0:d[i]));
    fwrite(r,1,l,stdout);
}