Cod sursa(job #1498755)

Utilizator StarGold2Emanuel Nrx StarGold2 Data 9 octombrie 2015 00:53:01
Problema PScNv Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <cstdio>
#include <vector>
#include <bitset>
#include <algorithm>

#define DIM 256000
using namespace std;

int N, M, st, fi, X, Y, Z;
vector <pair <int, int> > List[DIM];
bitset <DIM> Marked;

void DFS (int nod, int cost)
{
    Marked[nod] = 1;

    vector <pair <int, int> > :: iterator i;
    for (i = List[nod].begin(); i != List[nod].end(); i ++)
    {
        pair <int, int> vec = *i;

        if(!Marked[vec.first] && vec.second <= cost)
            DFS(vec.first, cost);
    }

    return;
}

int main()
{
    freopen ("pscnv.in" ,"r", stdin );
    freopen ("pscnv.out","w", stdout);

    scanf ("%d %d", &N , &M );
    scanf ("%d %d", &st, &fi);

    for (int i = 1; i <= M; i ++)
    {
        scanf ("%d %d %d", &X, &Y, &Z);
        List[X].push_back(make_pair(Y, Z));
    }

    int st = 1, dr = 1000, mid;
    while (st <= dr)
    {
        mid = st + ((dr - st) >> 1);
        Marked.reset();

        DFS (st, mid);

        if (Marked[fi])
            dr = mid - 1;
        else
            st = mid + 1;
    }

    printf ("%d\n", st);

    fclose (stdin );
    fclose (stdout);

    return 0;
}