Cod sursa(job #2465184)

Utilizator BalaBossBalaBoss BalaBoss Data 29 septembrie 2019 15:59:30
Problema Sate Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <bits/stdc++.h>
#define DIM 30005
#define INF 1e9
using namespace std;
ifstream fin ("sate.in");
ofstream fout ("sate.out");

    vector<pair<int,int>>g[DIM];
    int d[DIM],v[DIM];
    int n,m,a,b,c,x,y;
    queue<int> q;
void bfs()
{
    for(int i=1;i<=n;++i)
        d[i]=INF;
    d[x]=0;
    q.push(x);v[x]=1;
    while(!q.empty())
    {
        int nod=q.front();
        q.pop();
        for(auto w : g[nod])
        {
           int vecin=w.first;
           int cost=w.second;
           if(v[vecin]==0)
           {
            q.push(vecin);
            v[vecin]=1;
            if(vecin<nod)
                d[vecin]=d[nod]-cost;
            else if(vecin>nod)
                d[vecin]=d[nod]+cost;
            if(vecin==y)
                {fout<<d[y];return;}
            }
        }
    }
    fout<<d[y]<<" ";
}

int main()
{
    fin >>n >>m >>x>>y;
    for(int i=1;i<=m;++i)
    {
        fin>>a>>b>>c;
        g[a].push_back(make_pair(b,c));
        g[b].push_back(make_pair(a,c));
    }
    bfs();

    return 0;
}