Cod sursa(job #3370750)

Utilizator carinaa_leetAmarie Carina carinaa_leet Data 27 octombrie 2026 10:57:42
Problema Sate Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("sate.in");
ofstream fout("sate.out");

int n, m;
vector<vector<pair<int, int>>> adj;
vector<int> viz;

int BFS(int src, int fin)
{
    queue<int> q;
    q.push(src);
    viz[src] = 1;
    int dist_fin = 0;

    while (!q.empty())
    {
        int nod = q.front();
        q.pop();

        for (auto &vecin : adj[nod])
        {
            int i = vecin.first;
            int cost = vecin.second;
            if (!viz[i])
            {
                viz[i] = 1;
                if (i > nod)
                    dist_fin += cost;
                else
                    dist_fin -= cost;
                if (i == fin)
                {
                    return dist_fin;
                }
            }
            q.push(i);
        }
    }
    return 0;
}

int main()
{
    int x, y;
    fin >> n >> m >> x >> y;
    adj.resize(n + 5);
    viz.resize(n + 5);

    for (int i = 1; i <= n; i++)
    {
        viz[i] = 0;
    }

    for (int i = 0; i < m; i++)
    {
        int a, b, d;
        fin >> a >> b >> d;
        adj[a].push_back({b, d});
        adj[b].push_back({a, d});
    }

    fout << BFS(x, y);
    return 0;
}