Cod sursa(job #778857)

Utilizator visanrVisan Radu visanr Data 15 august 2012 23:55:16
Problema Sate Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.3 kb
#include <fstream>
#include <cstdlib>
#include <vector>
using namespace std;


#define pb push_back
#define mp make_pair
#define to first
#define cost second

vector<pair<int, int> > G[30010];
int N, M, X, Y, A, B, C, used[30010];

void DFS(int start, int dist);

int main()
{
    ifstream in("sate.in");
    ofstream out("sate.out");
    int i;
    in >> N >> M >> X >> Y;
    for(i = 1; i <= M; i++)
    {
          in >> A >> B >> C;
          G[A].pb(mp(B, C));
          G[B].pb(mp(A, C));
    }
    DFS(X, 0);
    return 0;
}

void DFS(int start, int dist)
{
     if(start == Y) 
     {
              out << dist << " ";
              exit(0);
     }
     used[start] = 1;
     vector<pair<int, int> > :: iterator it;
     for(it = G[start].begin(); it != G[start].end(); ++ it)
     {
        if(!used[it -> to])
        {
            if(it -> to < start)
            {
                   dist -= it -> cost;
                   DFS(it -> to, dist);
                   dist += it -> cost;
            }else
            {
                 if(it -> to > start)
                 {
                       dist += it -> cost;
                       DFS(it -> to, dist);
                       dist -= it -> cost;
                 }
            }
        }
     }
}