#include<bits/stdc++.h>
using namespace std;
using int64 = int64_t;
#define REP(i, n) for(int i = 0, _n = n; i < _n; ++i)
#define REPD(i, n) for(int i = n - 1; i >= 0; --i)
#define FOR(i, l, r) for(int i = l, _r = r; i <= _r; ++i)
#define FORD(i, r, l) for(int i = r, _l = l; i >= _l; --i)
#define left __left
#define right __right
#define prev __prev
#define next __next
#define div __div
#define pb push_back
#define pf push_front
#define sz(v) (int)v.size()
#define range(v) begin(v), end(v)
#define compact(v) v.erase(unique(range(v)), end(v))
#define debug(v) "[" #v " = " << (v) << "]"
template<typename T>
bool minimize(T& a, const T& b){
if(a > b){
a = b;
return true;
}
return false;
}
template<typename T>
bool maximize(T& a, const T& b){
if(a < b){
a = b;
return true;
}
return false;
}
template<int dimension, class T>
struct vec : public vector<vec<dimension - 1, T>> {
static_assert(dimension > 0, "Dimension must be positive !\n");
template<typename... Args>
vec(int n = 0, Args... args) : vector<vec<dimension - 1, T>>(n, vec<dimension - 1, T>(args...)) {}
};
template<class T>
struct vec<1, T> : public vector<T> {
vec(int n = 0, T val = T()) : vector<T>(n, val) {}
};
const int MAXN = 2e5 + 5;
const int MAXM = 4e5 + 5;
int n, m, q, number_of_nodes, tin[MAXN + MAXM], tout[MAXN + MAXM];
int lab[MAXN + MAXM], human[MAXN], werewolf[MAXN];
int pos_human[MAXN], pos_werewolf[MAXN], a[MAXN];
int S[MAXN], E[MAXN], L[MAXN], R[MAXN];
int l1[MAXN], r1[MAXN], l2[MAXN], r2[MAXN];
vector<int> adj[MAXN], g[MAXN + MAXM], order, ask[MAXN], st[MAXN << 2];
int find_root(int u){
return lab[u] < 0 ? u : (lab[u] = find_root(lab[u]));
}
bool unite(int u, int v){
u = find_root(u);
v = find_root(v);
if(u == v) return false;
g[number_of_nodes].pb(u);
if(u != v) g[number_of_nodes].pb(v);
lab[number_of_nodes] = lab[u] + lab[v];
lab[u] = lab[v] = number_of_nodes;
++number_of_nodes;
}
void dfs_euler(int u){
tin[u] = sz(order);
for(int v : g[u]){
dfs_euler(v);
}
if(u < n) order.pb(u);
tout[u] = sz(order) - 1;
}
void reset(){
order.clear();
number_of_nodes = n;
REP(i, n + m){
g[i].clear();
tin[i] = 0;
tout[i] = 0;
lab[i] = -1;
}
}
void process_human(){
reset();
REP(i, q){
ask[L[i]].pb(i);
}
REPD(i, n){
for(int j : adj[i]){
if(j >= i){
unite(i, j);
}
}
for(int id : ask[i]){
human[id] = find_root(S[id]);
}
ask[i].clear();
}
dfs_euler(number_of_nodes - 1);
REP(i, n){
pos_human[i] = tin[i];
}
REP(i, q){
l1[i] = tin[human[i]];
r1[i] = tout[human[i]];
}
}
void process_werewolf(){
reset();
REP(i, q){
ask[R[i]].pb(i);
}
REP(i, n){
for(int j : adj[i]){
if(j <= i){
unite(i, j);
}
}
for(int id : ask[i]){
werewolf[id] = find_root(E[id]);
}
ask[i].clear();
}
dfs_euler(number_of_nodes - 1);
REP(i, n){
pos_werewolf[i] = tin[i];
}
REP(i, q){
l2[i] = tin[werewolf[i]];
r2[i] = tout[werewolf[i]];
}
}
vector<int> combine(const vector<int>& a, const vector<int>& b){
vector<int> c;
int i = 0, j = 0;
while(i < sz(a) and j < sz(b)){
if(a[i] < b[j]) c.pb(a[i++]);
else c.pb(b[j++]);
}
for(; i < sz(a); ++i) c.pb(a[i]);
for(; j < sz(b); ++j) c.pb(b[j]);
return c;
}
void build(int id, int l, int r){
if(l == r) st[id] = {a[l]};
else{
int mid = l + r >> 1;
build(id << 1, l, mid);
build(id << 1 | 1, mid + 1, r);
st[id] = combine(st[id << 1], st[id << 1 | 1]);
}
}
int find_search(int u, int v, int at_least, int id = 1, int l = 0, int r = n - 1){
if(v < l or u > r) return INT_MAX;
if(u <= l && r <= v){
auto iter = lower_bound(range(st[id]), at_least);
return (iter == st[id].end() ? INT_MAX : *iter);
}
int mid = l + r >> 1;
return min(find_search(u, v, at_least, id << 1, l, mid), find_search(u, v, at_least, id << 1 | 1, mid + 1, r));
}
vector<int> check_validity(int N, vector<int> X, vector<int> Y, vector<int> _S, vector<int> _E, vector<int> _L, vector<int> _R){
n = N;
m = sz(X);
q = sz(_S);
REP(i, m){
int u = X[i], v = Y[i];
adj[u].pb(v);
adj[v].pb(u);
}
REP(i, q){
tie(S[i], E[i], L[i], R[i]) = make_tuple(_S[i], _E[i], _L[i], _R[i]);
}
process_human();
process_werewolf();
REP(i, n){
a[pos_human[i]] = pos_werewolf[i];
}
build(1, 0, n - 1);
vector<int> ans(q);
REP(i, q){
if(find_search(l1[i], r1[i], l2[i]) <= r2[i]){
ans[i] = 1;
}
else ans[i] = 0;
}
return ans;
}
//#define Zero_OP //turn off when submitting
#ifdef Zero_OP
void init(void);
void process(void);
int main(){
ios_base::sync_with_stdio(0); cin.tie(0);
#define task "antuvu"
if(fopen(task".inp", "r")){
freopen(task".inp", "r", stdin);
freopen(task".out", "w", stdout);
}
int T = 1; //cin >> T;
while(T--) {
init();
process();
}
return 0;
}
void init(){
}
void process(){
vector<int> ans = check_validity(6, {5, 1, 1, 3, 3, 5}, {1, 2, 3, 4, 0, 2},
{4, 4, 5}, {2, 2, 4}, {1, 2, 3}, {2, 2, 4});
REP(i, sz(ans)) cout << ans[i] << ' '; cout << '\n';
}
#endif
Compilation message
werewolf.cpp: In function 'void build(int, int, int)':
werewolf.cpp:171:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
171 | int mid = l + r >> 1;
| ~~^~~
werewolf.cpp: In function 'int find_search(int, int, int, int, int, int)':
werewolf.cpp:185:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
185 | int mid = l + r >> 1;
| ~~^~~
werewolf.cpp: In function 'bool unite(int, int)':
werewolf.cpp:74:5: warning: control reaches end of non-void function [-Wreturn-type]
74 | ++number_of_nodes;
| ^~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
49 ms |
96748 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
49 ms |
96748 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
190 ms |
150900 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
49 ms |
96748 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
