Cod sursa(job #492176)

Utilizator BitOneSAlexandru BitOne Data 13 octombrie 2010 17:58:10
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.32 kb
/* 
 * File:   main.cpp
 * Author: bitone
 *
 * Created on October 13, 2010, 5:33 PM
 */
#include <queue>
#include <fstream>
#include <cstdlib>
#include <iterator>
#include <algorithm>
#define MAX_N 50011
#define oo 1000000000

using namespace std;
typedef pair< int, int > pr;
/*
 * 
 */
int d[MAX_N];
bool was[MAX_N];
vector< pr > G[MAX_N];
vector< pr >::const_iterator it, iend;
class Compare
{
public :
    inline bool operator() ( const int& x, const int& y ) const
    {
        return d[x] > d[y];
    }
};
priority_queue< int, vector<int>, Compare > pQ;
int main( void )
{
    int N, M, x, y, c;
    ifstream in( "dijkstra.in" );
    for( in>>N>>M; M; --M )
    {
        in>>x>>y>>c;
        G[x].push_back( pr( y, c ) );
    }
    fill( d+2, d+N+1, oo );
    for( pQ.push(1); !pQ.empty(); pQ.pop() )
    {
        x=pQ.top();
        was[x]=false;
        for( it=G[x].begin(), iend=G[x].end(); it < iend; ++it )
            if( d[it->first] > d[x]+it->second )
            {
                d[it->first]=d[x]+it->second;
                if( false == was[it->first] )
                {
                    pQ.push(it->first);
                    was[it->first]=true;
                }
            }
    }
    ofstream out( "dijkstra.out" );
    copy( d+2, d+N+1, ostream_iterator<int>( out, " " ) );
    out<<'\n';
    return EXIT_SUCCESS;
}