Cod sursa(job #1391883)

Utilizator alexblackFMI - Dumitrache Alexandru alexblack Data 18 martie 2015 11:16:21
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.44 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 short n,m,x,y;
bool viz[N];
unsigned long cost[N];
queue<unsigned short> q;
struct element{unsigned short n; unsigned long c;};
vector<element> v[N];
void bfs()
{
    unsigned int cur=q.front();
    q.pop();
    int dim=v[cur].size();
    for(size_t i=0;((i<dim) && (!viz[y]));i++)
    {
        unsigned short a=v[cur][i].n;
        unsigned long cst=v[cur][i].c;
        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 unsigned long minmod(unsigned long a, unsigned long b)
{
    if(a>b) return a-b;
    return b-a;
}
int main()
{
    unsigned short a,b;
    unsigned long d;
    fscanf(in,"%hu %hu %hu %hu",&n,&m,&x,&y);
    for(unsigned i=1;i<=m;i++)
    {
        fscanf(in,"%hu %hu %lu\n",&a,&b,&d);
        element nou;
        nou.c=d;
        nou.n=b;
        v[a].push_back(nou);
        nou.n=a;
        v[b].push_back(nou);
        //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,"%lu\n",minmod(cost[x],cost[y]));
    return 0;
}