Cod sursa(job #2398372)

Utilizator StasBrega Stanislav Stas Data 5 aprilie 2019 13:25:36
Problema Sate Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <bits/stdc++.h>

using namespace std;

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

const int NMax=30005;

queue <int> coada;

vector <pair <int,int> > a[NMax];

int N,M,X,Y,dist[NMax];

void citire()
{
    fin >> N >> M >> X >> Y;
    for(int i=1;i<=M;i++)
    {
        int x,y,d;
        fin >> x >> y >> d;
        a[x].push_back(make_pair(y,d));
        a[y].push_back(make_pair(x,d));
    }
    for(int i=1;i<=N;i++)
        dist[i]=NMax;
    dist[X]=0;
    coada.push(X);
}
void bfs()
{
    int nod,vecin,distanta;
    while(!coada.empty())
    {
        nod=coada.front();
        coada.pop();
        for(int i=0;i<a[nod].size();i++)
        {
            vecin=a[nod][i].first;
            distanta=a[nod][i].second;
            if(dist[vecin]==NMax)
            {
                if(vecin>nod) dist[vecin]=dist[nod]+distanta;
                else dist[vecin]=dist[nod]-distanta;
                coada.push(vecin);
            }
        }
    }
}
int main()
{

    citire();
    bfs();
    fout << dist[Y];

    return 0;

}