| # | Time | Username | Problem | Language | Result | Execution time | Memory | 
|---|---|---|---|---|---|---|---|
| 295532 | zoooma13 | Werewolf (IOI18_werewolf) | C++14 | 0 ms | 0 KiB | 
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "werewolf.h"
#include "grader.cpp"
using namespace std;
vector <vector <int>> adj;
vector <int> a ,pos;
int gt(int u ,int p=-1){
    a.push_back(u);
    if(adj[u].size() == 1 && ~p)
        return u;
    if(adj[u][0] != p)
        return gt(adj[u][0] ,u);
    return gt(adj[u][1] ,u);
}
int getlastzero(int l ,int r ,int q){
    for(int i=r; i>=l; i--) if(a[i] < q)
        return i;
    return -1;
}
int getfirsttwo(int l ,int r ,int q){
    for(int i=l; i<=r; i++) if(a[i] > q)
        return i;
    return -1;
}
vector<int> check_validity(int N, vector<int> X, vector<int> Y,vector<int> S, vector<int> E,vector<int> L, vector<int> R) {
    assert(X.size() == N-1);
    adj.resize(N);
    for(int i=0; i<X.size(); i++){
        adj[X[i]].push_back(Y[i]);
        adj[Y[i]].push_back(X[i]);
    }
    int lf = gt(0); a.clear(); gt(lf);
    pos.resize(N);
    for(int i=0; i<N; i++)
        pos[a[i]] = i;
    ///bld
    vector <int> ret(R.size());
    for(int q = 0; q < R.size(); q++){
        int l = pos[S[q]] ,r = pos[E[q]];
        if(r < l) continue;
        int last0 = getlastzero(l ,r ,L[q]);
        int first2 = getfirsttwo(l ,r ,R[q]);
        if(last0==-1&&first2==-1) ret[q] = 1;
        else if(last0 == -1)      ret[q] = first2 != l;
        else if(first2 == -1)     ret[q] = 1;
        else                      ret[q] = (first2 >= last0+2);
    }
    reverse(a.begin() ,a.end());
    for(int i=0; i<N; i++)
        pos[a[i]] = i;
    ///bld
    for(int q = 0; q < R.size(); q++){
        int l = pos[S[q]] ,r = pos[E[q]];
        if(r < l) continue;
        int last0 = getlastzero(l ,r ,L[q]);
        int first2 = getfirsttwo(l ,r ,R[q]);
        if(last0==-1&&first2==-1) ret[q] = 1;
        else if(last0 == -1)      ret[q] = first2 != l;
        else if(first2 == -1)     ret[q] = 1;
        else                      ret[q] = (first2 >= last0+2);
    }
    return ret;
}
