Cod sursa(job #2151534)

Utilizator futurengineerOana Rosca futurengineer Data 4 martie 2018 16:39:41
Problema Sate Scor 35
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

struct vecin {
  int n, c;
};

int n, m, x, y, d[30001], sursa, destinatie;
bool viz[30001];
vector<vecin> vec[30001];
queue<int> q;

void bfs(int nod) {
  q.push(nod);
  while (!q.empty()) {
    int nodc = q.front(); q.pop(); viz[nodc] = 1;
    //vector<vecin>::iterator i;
    for (int i = 0; i < vec[nodc].size(); i++)
      if (!viz[vec[nodc][i].n]) {
        if (nodc < vec[nodc][i].n)
          d[vec[nodc][i].n] += d[nodc]+vec[nodc][i].c;
        else
          d[vec[nodc][i].n] += d[nodc]-vec[nodc][i].c;
        if (vec[nodc][i].n == destinatie)
          return;
        q.push(vec[nodc][i].n);
      }
  }
}

int main () {
  ifstream fi("sate.in");
  ofstream fo("sate.out");
  fi >> n >> m >> sursa >> destinatie;
  for (int i = 1; i <= m; i++) {
    vecin nod;
    fi >> x >> y >> nod.c;
    nod.n = y;
    vec[x].push_back(nod);
    nod.n = x;
    vec[y].push_back(nod);
  }
  bfs(sursa);
  fo << d[destinatie];
  return 0;
}