Cod sursa(job #1391827)

Utilizator alexblackFMI - Dumitrache Alexandru alexblack Data 18 martie 2015 10:42:58
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.15 kb
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
FILE *in=fopen("sate.in","r");
FILE *out=fopen("sate.out","w");
long const N=30001;
unsigned n,m,x,y;
bool viz[N];
unsigned long cost[N];
vector<unsigned> v[N];
vector<unsigned long> c[N];
queue<unsigned> q;
void bfs()
{
    int cur=q.front();
    q.pop();
    int dim=v[cur].size();
    for(size_t i=0;((i<dim) && (!viz[y]));i++)
    {
        int a=v[cur][i];
        int cst=c[cur][i];
        if(viz[a]) continue;
        q.push(a);
        viz[a]=true;
        if(a>cur)
            cost[a]=cost[cur]+cst;
        else
            cost[a]=cost[cur]-cst;
    }
}
inline int mod(int x)
{
    if(x>0) return x;
    return -x;
}
int main()
{
    fscanf(in,"%d %d %d %d",&n,&m,&x,&y);
    for(unsigned i=1;i<=m;i++)
    {
        int a,b,d;
        fscanf(in,"%d %d %d",&a,&b,&d);
        v[a].push_back(b);
        c[a].push_back(d);
        v[b].push_back(a);
        c[b].push_back(d);
    }
    q.push(x);
    viz[x]=true;
    bfs();
    while((!q.empty())&&(!viz[y]))
        bfs();
    fprintf(out,"%d\n",mod(cost[x]-cost[y]));
    return 0;
}