Cod sursa(job #1389360)

Utilizator alexblackFMI - Dumitrache Alexandru alexblack Data 16 martie 2015 10:43:41
Problema Sate Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
ifstream in ("sate.in");
ofstream out("sate.out");
int const N=20000;
int n,m,x,y;
bool viz[N];
int cost[N];
vector<int> v[N], c[N];
queue<int> q;
void bfs()
{
    int cur=q.front();
    q.pop();
    viz[cur]=true;
    for(int i=0;i<v[cur].size();i++)
    {
        int a=v[cur][i];
        int cst=c[cur][i];
        if(viz[a]) continue;
        q.push(a);
        if(a>cur)
            cost[a]=cost[cur]+cst;
        else
            cost[a]=cost[cur]-cst;
    }
}
void init()
{
    for(int i=0;i<v[1].size();i++)
        cost[v[1][i]]=c[1][i];
}
int mod(int x)
{
    if(x>0) return x;
    return -x;
}
int main()
{
    in>>n>>m>>x>>y;
    for(int i=1;i<=m;i++)
    {
        int a,b,d;
        in>>a>>b>>d;
        v[a].push_back(b);
        c[a].push_back(d);
        v[b].push_back(a);
        c[b].push_back(d);
    }
    init();
    q.push(1);
    bfs();
    while(!q.empty())
        bfs();
    out<<mod(cost[x]-cost[y])<<"\n";
    return 0;
}