Submission #1092458

#TimeUsernameProblemLanguageResultExecution timeMemory
1092458huyngoWerewolf (IOI18_werewolf)C++17
49 / 100
386 ms77332 KiB
#include "werewolf.h" #include <bits/stdc++.h> using namespace std; using ll = long long; using i64 = long long; void __print(int x) { cerr << x; } void __print(long x) { cerr << x; } void __print(long long x) { cerr << x; } void __print(unsigned x) { cerr << x; } void __print(unsigned long x) { cerr << x; } void __print(unsigned long long x) { cerr << x; } void __print(float x) { cerr << x; } void __print(double x) { cerr << x; } void __print(long double x) { cerr << x; } void __print(char x) { cerr << '\'' << x << '\''; } void __print(const char* x) { cerr << '"' << x << '"'; } void __print(const string& x) { cerr << '"' << x << '"'; } void __print(bool x) { cerr << (x ? "true" : "false"); } template<typename T, typename V> void __print(const pair<T, V>& x) { cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}'; } template<typename T> void __print(const T& x) { int f = 0; cerr << '{'; for (auto& i : x) cerr << (f++ ? "," : ""), __print(i); cerr << "}"; } void _print() { cerr << "]\n"; } template <typename T, typename... V> void _print(T t, V... v) { __print(t); if (sizeof...(v)) cerr << ", "; _print(v...); } #define dbg(x...) cerr << "[" << #x << "] = ["; _print(x) #define ln "\n" #define fastIO() ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0) #define all(x) (x).begin(), (x).end() #define rep(i,a,b) for(int i=a; i<=b; ++i) #define ar array int Bit(int mask, int b) { return (mask >> b) & 1; } const ll base = 311, MOD = 998244353, M = 1e9 + 7, INF = 1e18; std::vector<int> check_validity(int N, std::vector<int> X, std::vector<int> Y, std::vector<int> S, std::vector<int> E, std::vector<int> L, std::vector<int> R) { int Q = S.size(); int M = X.size(); std::vector<int> ANS(Q); vector<vector<int>> adj(N); rep(i, 0, M - 1) { adj[X[i]].push_back(Y[i]); adj[Y[i]].push_back(X[i]); } if (N <= 3000) { rep(i, 0, Q - 1) { vector<int> vis(N, 0); int ok = 0; auto dfs = [&](auto self, int u, int lb, int ub) -> void { if (u < lb || u > ub) return; vis[u] = (ub == 1e8 ? 1 : 2); for (int v : adj[u]) if (!vis[v]) { if (lb <= v && v <= ub) self(self, v, lb, ub); } else { if (lb == -1 && vis[v] == 1 && v <= R[i]) ok = 1; } }; dfs(dfs, S[i], L[i], 1e8); if (E[i] <= R[i] && vis[E[i]] == 1) ok = true; dfs(dfs, E[i], -1, R[i]); ANS[i] = ok; } return ANS; } vector<int> pos(N); int dtime = -1; auto dfs = [&](auto self, int u, int p) -> void { pos[u] = ++dtime; for (int v : adj[u]) if (v != p) self(self, v, u); }; rep(i, 0, N - 1) if (adj[i].size() == 1) { dfs(dfs, i, -1); break; } vector<int> a(N, 0); iota(all(a), 0); sort(all(a), [&](int i, int j) { return pos[i] < pos[j]; }); struct Info { int min = 1e9; int max = -1e9; Info operator + (const Info& p) { return Info{ std::min(min, p.min), std::max(max, p.max) }; } }; vector<vector<Info>> g(N, vector<Info>(21)); for (int i = 0; i < N; ++i) g[i][0] = { a[i], a[i] }; for (int j = 1; j < 21; ++j) for (int i = 0; i + (1 << j) <= N; ++i) g[i][j] = g[i][j - 1] + g[i + (1 << (j - 1))][j - 1]; auto get = [&](int l, int r) -> Info { int k = log2(r - l + 1); return g[l][k] + g[r - (1 << k) + 1][k]; }; rep(i, 0, Q - 1) { if (S[i] < L[i] || E[i] > R[i]) continue; if (pos[S[i]] < pos[E[i]]) { // min(pos[si], mid) >= L // max(mid, pos[ei]) <= R int lo = pos[S[i]], hi = pos[E[i]], md; int x = lo; while (lo <= hi) { md = (lo + hi) / 2; if (get(pos[S[i]], md).min >= L[i]) { lo = md + 1; x = max(x, md); } else hi = md - 1; } ANS[i] = get(x, pos[E[i]]).max <= R[i]; } else { // min(mid, pos[si]) >= L // max(pos[ei], mid) <= R int lo = pos[E[i]], hi = pos[S[i]], md; int x = hi; while (lo <= hi) { md = (lo + hi) / 2; if (get(md, pos[S[i]]).min >= L[i]) { hi = md - 1; x = min(x, md); } else lo = md + 1; } ANS[i] = get(pos[E[i]], x).max <= R[i]; } } return ANS; }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...