Cod sursa(job #1896608)

Utilizator nartorrewrew narto Data 28 februarie 2017 19:46:00
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <fstream>
#include <queue>
#include <vector>
#include <bitset>
#define inf 1<<30
#define nmax 100005
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");


int n, m, d[nmax],start;
vector<pair<unsigned short int,unsigned short int> >v[nmax];
vector<pair<unsigned short int,unsigned short int> >::iterator it;

bitset<nmax> uz;


class comp{

 public:
     bool operator()(pair<unsigned short int,unsigned short int> a,pair<unsigned short int,unsigned short int> b)
     {
         return a.first>b.first;
     }

};
priority_queue<pair<int,int>,vector<pair<int,int> >,comp >heap;

void dijsktra()
{ int x, i, y, z;
    for(i=1;i<=n;i++)
        d[i]=inf;
        d[1]=0;
    heap.push(make_pair(0,1));
    while(!heap.empty())
    {
        x=heap.top().second;
        heap.pop();

        if(uz[x]==1)
            continue;
        uz[x]=1;
        for(it=v[x].begin();it!=v[x].end();it++){
            y=it->first;
            z=it->second;
            if(d[y]>d[x]+z){
             d[y]=d[x]+z;
             heap.push(make_pair(d[y],y));
        }
    }
}
  for(i=2;i<=n;i++)
  {if(d[i]!=inf)
     g<<d[i]<<' ';
  else
    g<<0<<' ';}
}
int main()
{ int i, x, y, c;
    f>>n>>m;
    for(i=1;i<=m;i++)
    { f>>x>>y>>c;
      v[x].push_back(make_pair(y,c));
      //v[y].push_back(make_pair(x,c));
    }
    dijsktra();

}