# | 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;
}