Cod sursa(job #1438890)

Utilizator eneandradaEne Oana-Andrada eneandrada Data 21 mai 2015 01:23:45
Problema Sate Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.22 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define MAX_N 30005

using namespace std;

int dist[MAX_N], X, Y;
struct drum
{
    int nr_sat;
    int lung;
    drum(int nr, int l): nr_sat(nr), lung(l){};
};
vector <drum> G[MAX_N];

void BFS(int nod)
{
    queue <int> Q;
    Q.push(nod);
    while(!Q.empty())
    {
        int n = Q.front();
        for(int i = 0; i < G[n].size(); i++)
        {
            if(dist[G[n][i].nr_sat] == 0)
            {
                if(G[n][i].nr_sat > n)
                    dist[G[n][i].nr_sat] = dist[n] + G[n][i].lung;
                else
                    dist[G[n][i].nr_sat] = dist[n] - G[n][i].lung;
                Q.push(G[n][i].nr_sat);
                if(G[n][i].nr_sat == Y)
                    return;
            }
        }
        Q.pop();
    }
}


int main()
{
    int N, M, a, b, c, i;
    freopen("sate.in", "r", stdin);
    scanf("%d%d%d%d", &N, &M, &X, &Y);
    for(i = 1; i <= M; i++)
    {
        scanf("%d%d%d", &a, &b, &c);
        G[a].push_back(drum(b,c));
        G[b].push_back(drum(a,c));
    }
    BFS(X);
    freopen("sate.out", "w", stdout);
    printf("%d", dist[Y]);

    return 0;
}