Cod sursa(job #1394099)

Utilizator AlexTAlexandru AlexT Data 19 martie 2015 23:54:19
Problema Sate Scor 55
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <iostream>
#include <fstream>
#include <vector>
#include<queue>
using namespace std;
int main()
{
int k, j, i, n, m, b, c, x, y, d, vizitat[30001],parinte[30001];
pair<int,int> pereche;
vector< pair <int,int> > v[30001];
queue<int> q;
bool gasit=0;
ifstream f("sate.in");
ofstream g("sate.out");
f >> n >> m >> b >>c;
for(i=0;i<n;i++)
    {
    parinte[i]=vizitat[i]=0;
    }
for(i=0;i<m;i++)
    {
    f >> x >> y >> d;
    v[x].push_back(make_pair(y,d));
    v[y].push_back(make_pair(x,d));
    }

q.push(b);
while((!q.empty())&&(gasit==0))
{
k=q.front();
for(i=0;i<v[k].size();i++)
    {
    if(vizitat[v[k][i].first]==0)
        {
        parinte[v[k][i].first]=k;
        vizitat[v[k][i].first]=1;
        q.push(v[k][i].first);
        }
    if(v[k][i].first==c)
        {
        gasit=1;
        break;
        }
    }
q.pop();
}
k=c;
d=0;
while(k!=b)
    {
    for(i=0;i<v[k].size();i++)
        {
        if(v[k][i].first==parinte[k])
            {
            if(v[k][i].first<k)
                {
                d+=v[k][i].second;
                }
            else
                {
                d-=v[k][i].second;
                }
            }
        }
    k=parinte[k];
    }
g<<d;


f.close();
g.close();
    return 0;
}