Cod sursa(job #1878745)

Utilizator marcdariaDaria Marc marcdaria Data 14 februarie 2017 14:05:34
Problema Sate Scor 80
Compilator cpp Status done
Runda becreative15 Marime 0.77 kb
# include <fstream>
# include <vector>
# define f first
# define s second

using namespace std;

ifstream fin("sate.in");
ofstream fout("sate.out");
using PII = pair<int,int>;
vector<PII> G[30010];

int viz[30010], n, m, x, y, a, b, cost;

void DFS(int nod);

int main()
{

    fin>>n>>m>>x>>y;

    if(x>y) swap(x,y);

    for(int i=1; i<=m; i++){
        fin>>a>>b>>cost;
        G[a].push_back(make_pair(b, cost));
        G[b].push_back(make_pair(a, -cost));
    }

    DFS(x);

    fout<<viz[y]<<"\n";

    return 0;
}
void DFS(int nod)
{
    for(int i=0; i<G[nod].size(); i++){
        int fs=G[nod][i].f;
        int cost=G[nod][i].s;
        if(!viz[fs]){
            viz[fs]=viz[nod]+cost;
            DFS(fs);
        }
    }
}