Cod sursa(job #3216217)

Utilizator mmocanuMocanu Mihai-Adrian mmocanu Data 15 martie 2024 18:32:13
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.29 kb
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <queue>
#define MAXN 50004
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
using namespace std;

struct XYZ{
  int node,cost;
};
struct ABC{
  int neigh,cost;
};

bool operator<(XYZ a,XYZ b){
  return a.cost<b.cost;
}

vector <ABC> graf[MAXN];
priority_queue <XYZ> q;
int cost[MAXN];
int seen[MAXN];

void DFS(XYZ node){
  cost[node.node]=1;
  q.push(node);
  while(!q.empty()){
    node=q.top();
    q.pop();
    if(seen[node.node]==0){
      seen[node.node]=1;
      for(ABC neigh : graf[node.node]){
        if(cost[neigh.neigh]>cost[node.node]+neigh.cost || cost[neigh.neigh]==0){
          cost[neigh.neigh]=cost[node.node]+neigh.cost;
          q.push({neigh.neigh,cost[neigh.neigh]});
        }
      }
    }
  }
}

int main(){
  int n,m,x,y,z,i;
  FILE *fin,*fout;
  fin=fopen("dijkstra.in","r");
  fout=fopen("dijkstra.out","w");
  fscanf(fin,"%d%d",&n,&m);

  for(i=0;i<m;i++){
    fscanf(fin,"%d%d%d",&x,&y,&z);
    graf[x].push_back({y,z});
  }

  DFS({1,1});

  for(i=2;i<=n;i++){
    if(cost[i]==0){
      cost[i]=1;
    }
    fprintf(fout,"%d ",cost[i]-1);
  }
  fprintf(fout,"\n");

  fclose(fin);
  fclose(fout);
  return 0;
}