Cod sursa(job #46764)

Utilizator fireatmyselfBogdan-Alexandru Stoica fireatmyself Data 2 aprilie 2007 22:19:08
Problema PScNv Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.84 kb
#include <cstdio>
#include <cstring>
#include <vector>
#include <cstdlib>
#define NMAX 200200
#define CMAX 200020
#define max(a, b) ((a) > (b) ? (a):(b))

using namespace std;

vector<int> A[NMAX];
vector<int> K[NMAX];
int N, M, X, Y, Kmin, F[NMAX], C[CMAX+10];
char buf[NMAX*40];

int main()
{
        int i, j, c, k, hi, lo, mij, m = 0, p1, p2, ok, nod;

        FILE *f = fopen("pscnv.in", "rt");
        setbuffer(f,buf,sizeof(buf));
        fscanf(f,"%d %d %d %d", &N, &M, &X, &Y);

        while (M--)
        {
                fscanf(f,"%d %d %d", &i, &j, &c);
                A[i].push_back(j);
                A[j].push_back(i);
                K[i].push_back(c);
                K[j].push_back(c);
                m = max(m, c);
        }

        lo = 0; hi = m;

        while (hi-lo >= 0)
        {
                mij = (hi+lo)>>1;
                memset(C, 0, sizeof(C));
                C[p1 = 0] = X; p2 = 1;
                memset(F, 0, sizeof(F));
                F[X] = 1;
                ok = 0;
                while (p2 != p1 && ok == 0)
                {
                        nod = C[p1];
                        m = A[nod].size();
                        for (i = 0; i < m && ok == 0; i++)
                            if (F[ A[nod][i] ] == 0 && K[nod][i] <= mij)
                            {
                                F[ A[nod][i] ] = 1;
                                C[p2] = A[nod][i];
                                ok = (A[nod][i] == Y);
                                p2 = (p2+1)%CMAX;
                            }
                        p1 = (p1+1)%CMAX;
                }
                if (ok == 1) hi = mij-1, Kmin = mij;
                else lo = mij+1;
        }

        freopen("pscnv.out", "w", stdout);
        printf("%d\n", Kmin);

        return 0;
        
}