Cod sursa(job #1605534)

Utilizator RobertMMinzat Robert RobertM Data 19 februarie 2016 09:26:53
Problema Algoritmul lui Dijkstra Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <fstream>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
#define inf (1<<15)
int n,c[5001][5001],d[5001],t[5001],s[5001],xp=1,m;
void cit(){
    int i,j,x,y,z;
    fin>>n>>m;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            c[i][j]=inf;
    while(fin){
        fin>>x>>y>>z;
        c[x][y]=z;}
}
int min(int &q){
    int i,m=2*inf;
    for(i=1;i<=n;i++)
        if(!s[i] && d[i]<m){
            m=d[i];
            q=i;}
}
void det_drumuri(){
    int i,x,j,k,ok;
    for(i=1;i<=n;i++){
        d[i]=c[xp][i];
        s[i]=0;
        if(c[xp][i]<inf)
            t[i]=xp;
        else
            t[i]=0;}
    s[xp]=1;
    t[xp]=0;
    ok=1;
    x=0;
    do{ min(k);
        x++;
        if(d[k]==inf || x==n)
            ok=0;
        else{
            s[k]=1;
            for(j=1;j<=n;j++)
                if(!s[j] && d[j]>d[k]+c[k][j]){
                    d[j]=d[k]+c[k][j];
                    t[j]=k;}
}
    }while(ok);
}

int main()
{
    cit();
    det_drumuri();
    for(int i=2;i<=n;i++)
	if(d[i]==inf)
	fout<<0<<" ";
	else
        fout<<d[i]<<" ";
    return 0;
}