Cod sursa(job #2798414)

Utilizator mihneadv@yahoo.comDavid Mihnea Stefan [email protected] Data 11 noiembrie 2021 11:46:52
Problema Sate Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.56 kb
#include<fstream>
#include<vector>
#include<queue>
#define DMAX 30000

using namespace std;

ifstream cin("sate.in");
ofstream cout("sate.out");

int suma[DMAX], n, dr, drum[DMAX];
struct el
{
    int vecin, lung;
};
vector<el>v[DMAX];

void BFS(int start)
{
    queue<int>Q;
    Q.push(start);
    suma[start] = 1;
    while(!Q.empty())
    {
        int actual = Q.front();
        Q.pop();
        for(int i = 0; i < v[actual].size(); i++)
        {
            if(suma[v[actual][i].vecin] == false )
            {
                if(v[actual][i].vecin > actual)
                {
                    suma[v[actual][i].vecin] = suma[actual] + v[actual][i].lung;
                }
                else
                {
                    suma[v[actual][i].vecin] = suma[actual] - v[actual][i].lung;
                }
                Q.push(v[actual][i].vecin);
            }
        }
    }
}

int main()
{
    int m, st;
    cin >> n >> m >> st >> dr;
    for(int i = 0; i < m; i++)
    {
        int V1, V2, d;
        cin >> V1 >> V2 >> d;
        V1--, V2--;
        el p;
        p.vecin = V2;
        p.lung = d;
        v[V1].push_back(p);
        p.vecin = V1;
        p.lung = d;
        v[V2].push_back(p);
    }
    /*
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < v[i].size();i++)
        {
            cout << i <<  " " << v[i][j].vecin << " " << v[i][j].lung << "     ";
        }
        cout << '\n';
    }*/
    st--, dr--;
    BFS(st);
    cout << suma[dr] - 1;
    return false;
}