Cod sursa(job #1500921)

Utilizator StarGold2Emanuel Nrx StarGold2 Data 12 octombrie 2015 21:28:15
Problema PScNv Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.05 kb
#include <cstdio>
#include <vector>
#include <bitset>
#include <algorithm>

#define DIM 251000
using namespace std;

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

class InputReader {
    public:
        InputReader() {}
        InputReader(const char *file_name) {
            input_file = fopen(file_name, "r");
            cursor = 0;
            fread(buffer, SIZE, 1, input_file);
        }
        inline InputReader &operator >>(int &n) {
            while(buffer[cursor] < '0' || buffer[cursor] > '9') {
                advance();
            }
            n = 0;
            while('0' <= buffer[cursor] && buffer[cursor] <= '9') {
                n = n * 10 + buffer[cursor] - '0';
                advance();
            }
            return *this;
        }
    private:
        FILE *input_file;
        static const int SIZE = 1 << 17;
        int cursor;
        char buffer[SIZE];
        inline void advance() {
            ++ cursor;
            if(cursor == SIZE) {
                cursor = 0;
                fread(buffer, SIZE, 1, input_file);
            }
        }
};

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()
{
    InputReader fin("pscnv.in" );
    freopen ("pscnv.out","w", stdout);

    fin >> N >> M;
    fin >> st >> fi;

    for (int i = 1; i <= M; i ++)
    {
        fin >> X >> Y >> Z;
        List[X].push_back(make_pair(Y, Z));
    }

    int l = 1, r = 1000, mid;
    while (l <= r)
    {
        mid = l + ((r - l) >> 1);

        DFS (st, mid);

        if (Marked[fi])
            r = mid - 1;
        else
            l = mid + 1;

        Marked.reset();
    }

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

    fclose (stdin );
    fclose (stdout);

    return 0;
}