Cod sursa(job #2257573)

Utilizator serban24Popovici Serban-Florin serban24 Data 10 octombrie 2018 10:45:01
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <bits/stdc++.h>

using namespace std;

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

vector < pair <int,int> > mv[30005];
queue < pair <int,int> > q;
int viz[30005];
int val[30005];
int n,m,x,y,nr;

void BFS(int v){
    int i;

    viz[v]=1;
    q.push(make_pair(v,0));

    while(!q.empty()){
        v=q.front().first;
        q.pop();

        for(i=0;i<mv[v].size();i++){
            if(!viz[mv[v][i].first]){
                viz[mv[v][i].first]=1;

                if(mv[v][i].first>v)
                    val[mv[v][i].first]=val[v]+mv[v][i].second;
                else
                    val[mv[v][i].first]=val[v]-mv[v][i].second;

                q.push(make_pair(mv[v][i].first,val[mv[v][i].first]));
            }
        }
    }
}

int main(){
    int i,a,b;

    fin>>n>>m>>x>>y;

    for(i=1;i<=m;i++){
        fin>>a>>b>>nr;

        mv[a].push_back(make_pair(b,nr));
        mv[b].push_back(make_pair(a,nr));
    }

    BFS(x);

    fout<<val[y];

    return 0;
}