Cod sursa(job #2668660)

Utilizator jucatorulGrigore George Alexandru jucatorul Data 5 noiembrie 2020 07:52:10
Problema Sate Scor 100
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2013 Marime 0.95 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;



int sol[30001], vizitat[30001];
vector <pair<int,int>> vecin[30001];
queue <int> q;

void bfs(int k)
{

    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        for(auto i : vecin[x])
        {
            if(!vizitat[i.first])
            {
                q.push(i.first);
                vizitat[i.first]=1;
                if(x > i.first)
                    sol[i.first]=sol[x]-i.second;
                else
                    sol[i.first]=sol[x]+i.second;

            }
        }
    }
}

int main()
{ifstream f("sate.in");
ofstream g("sate.out");
int n,m,x,y,d,i,j;
    f>>n>>m>>x>>y;
    for(int k=1;k<=m;k++)
    {
        f>>i>>j>>d;
        vecin[i].push_back({j, d});
        vecin[j].push_back({i, d});
    }

    q.push(x);
    vizitat[x]=1;
    bfs(x);
    g<<sol[y];

    return 0;
}