Cod sursa(job #1076952)

Utilizator IoannaPandele Ioana Ioanna Data 10 ianuarie 2014 19:11:26
Problema Sate Scor 80
Compilator cpp Status done
Runda Teme Pregatire ACM Unibuc 2013 Marime 1.21 kb
#include <fstream>
#include <queue>
#include <vector>
#define NMAX 30002
using namespace std;

int n, m;
int x, y;
int dist;
queue<int> q;
queue<int> d;
vector<int> v[NMAX];
vector<int> c[NMAX];
char viz[NMAX];


ifstream in("sate.in");
ofstream out("sate.out");

void read() {
    in>>n>>m>>x>>y;
    int a, b, cost;
    if (x > y) swap(x, y);

    for (int i = 1; i <= m; i++) {
        in>>a>>b>>cost;
        v[a].push_back(b);
        c[a].push_back(cost);
        v[b].push_back(a);
        c[b].push_back(cost);
    }
}

void bfs() {
    q.push(x);
    d.push(0);
    viz[x] = 1;
    while (!q.empty()) {
        int sat = q.front();
        dist = d.front();
        if (sat == y)
            break;
        q.pop();
        d.pop();
        for (int i = 0; i < v[sat].size(); i++) {
            if (!viz[v[sat][i]]) {
                viz[v[sat][i]] = 1;
                q.push(v[sat][i]);
                if (v[sat][i] > sat) {
                    d.push(dist + c[sat][i]);
                } else {
                    d.push(dist - c[sat][i]);
                }
            }
        }
    }
}

int main() {
    read();
    bfs();
    out<<dist;
    return 0;
}