Cod sursa(job #1982448)

Utilizator dragos231456Neghina Dragos dragos231456 Data 18 mai 2017 20:20:39
Problema Sate Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <deque>
using namespace std;
ifstream f("sate.in");
ofstream g("sate.out");
vector<int> vecini[30005],distanta[30005];
deque<int> q;
string s;
int n,m,x,y,a,b,start,d,cost[30005],deparcurs[30005],c;

int next(int &c)
{
    int nr=0,cif;
    while(s[c]!=' ' && c<s.size())
    {
        cif=s[c]-'0';
        nr=nr*10+cif;
        ++c;
    }
    ++c;
    return nr;
}

void bfs(int x)
{
    q.push_front(x);
    while(!q.empty())
    {
        start=q.front();
        q.pop_front();
        for(int i=0;i<deparcurs[start];++i)
        {
            if(cost[vecini[start][i]]==0)
            {
                if(vecini[start][i]<start) cost[vecini[start][i]]=cost[start]-distanta[start][i];
                else cost[vecini[start][i]]=cost[start]+distanta[start][i];
                if(vecini[start][i]==y) break;
                else q.push_back(vecini[start][i]);
            }
        }
    }
}

int main()
{
    f>>n>>m>>x>>y; getline(f,s);
    for(int i=1;i<=m;++i)
    {
        getline(f,s);
        c=0;
        a=next(c); b=next(c); d=next(c);
        vecini[a].push_back(b); vecini[b].push_back(a);
        distanta[a].push_back(d); distanta[b].push_back(d);
    }
    for(int i=1;i<=n;++i)
    {
        deparcurs[i]=vecini[i].size();
    }
    bfs(x);
    g<<cost[y];
    return 0;
}