Cod sursa(job #2530755)

Utilizator CristiVintilacristian vintila CristiVintila Data 25 ianuarie 2020 11:50:24
Problema Sate Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define NMAX 30005
using namespace std;
ifstream fin("sate.in");
ofstream fout("sate.out");
 
int n, m, x, y, sum, viz[NMAX];
vector< pair<int, int> > gr[NMAX];
queue<int> q;

void citire() {
  fin >> n >> m >> x >> y;
  for (int i = 1; i <= m; i++) {
    int a, b, cost;
    fin >> a >> b >> cost;
    gr[a].push_back(make_pair(b, cost));
    gr[b].push_back(make_pair(a, -1 * cost));
  }
}
 
void BFS(int n_start, int n_end) {
  for (int i = n_start; i <= n_end; i++)
    viz[i] = -1;
  
  viz[n_start] = 0;
  q.push(n_start);
  
  while (!q.empty()) {
    int nodC = q.front();
    q.pop();
    for (int i = 0; i < gr[nodC].size(); i++) {
      int comp = gr[nodC][i].first;
      int val = gr[nodC][i].second;
      if (viz[comp] == -1) {
        viz[comp] = viz[nodC] + val;
        if (comp == n_end) {
          fout << viz[n_end];
          return;
        }
        q.push(comp);
      }
    }
  }
}
 
int main(int argc, const char * argv[]) {
  citire();
  BFS(x, y);
}