/**
* author: NimaAryan
* created: 2024-01-22 15:04:53
**/
#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template <typename T, class C = less<>>
using indexed_set = tree<T, null_type, C, rb_tree_tag,
tree_order_statistics_node_update>;
#ifdef LOCAL
#include "algo/debug.h"
#endif
using i64 = long long;
struct Solver {
int n, k;
vector<vector<pair<int, int>>> adj;
vector<vector<pair<int, int>>> qry1;
vector<vector<int>> qry2;
vector<int> siz;
vector<bool> alive;
vector<int> join;
vector<int> leave;
vector<int> reach;
vector<int> label;
vector<int> ans;
Solver(int n, int k) : n(n), k(k) {
adj.resize(n);
qry1.resize(n);
qry2.resize(n);
siz.resize(n);
alive.assign(n, true);
join.assign(n, -1);
leave.assign(n, -1);
reach.assign(n, -1);
label.resize(n);
ans.resize(n - 1 + k);
}
void add_edge(int a, int b, int w) {
adj[a].emplace_back(b, w);
adj[b].emplace_back(a, w);
}
void add_query1(int a, int d, int t) {
qry1[d].emplace_back(a, t);
}
void add_query2(int d, int t) {
qry2[d].push_back(t);
}
void dfs(int v, int p = -1) {
siz[v] = 1;
for (auto [u, _] : adj[v]) {
if (u != p && alive[u]) {
dfs(u, v);
siz[v] += siz[u];
}
}
leave[v] = -1;
join[v] = reach[v] = n + k;
}
int find_centroid(int v, int z, int p = -1) {
for (auto [u, _] : adj[v]) {
if (u != p && alive[u] && siz[u] > z / 2) {
return find_centroid(u, z, v);
}
}
return v;
}
void solve(int v = 0) {
dfs(v);
int c = find_centroid(v, siz[v]);
{
/* increasing */
auto downward = [&](auto self, int x, int p) -> void {
for (auto [y, z] : adj[x]) {
if (y != p && alive[y] && reach[x] < z) {
leave[y] = leave[x];
reach[y] = z;
self(self, y, x);
}
}
};
for (auto [u, w] : adj[c]) {
if (alive[u]) {
leave[u] = reach[u] = w;
downward(downward, u, c);
}
}
}
{
/* decreasing */
auto upward = [&](auto self, int x, int w, int p) -> void {
for (auto [y, z] : adj[x]) {
if (y != p && alive[y] && w > z) {
join[y] = join[x];
self(self, y, z, x);
}
}
};
for (auto [u, w] : adj[c]) {
if (alive[u]) {
join[u] = w;
upward(upward, u, w, c);
}
}
}
sort(adj[c].begin(), adj[c].end(), [&](auto p, auto q) {
return p.second > q.second;
});
vector<vector<int>> subs;
auto trav = [&](auto self, int x, int p) -> void {
subs.back().push_back(x);
label[x] = subs.size();
for (auto [y, _] : adj[x]) {
if (y != p && alive[y]) {
self(self, y, x);
}
}
};
for (auto [u, _] : adj[c]) {
if (alive[u]) {
subs.emplace_back();
trav(trav, u, c);
}
}
indexed_set<int> can;
for (auto s : subs) {
for (int d : s) {
for (auto [a, t] : qry1[d]) {
if (ans[t]) {
continue;
}
if (a == d) {
ans[t] = -1;
continue;
}
if (a == c) {
ans[t] = -1 + -(join[d] >= t);
continue;
}
if (label[a] == label[d]) {
continue;
}
ans[t] = -1 + (join[d] >= leave[a] || reach[a] >= t);
}
for (int t : qry2[d]) {
if (join[d] < t) {
ans[t] += can.order_of_key(t) + 1;
}
}
}
for (int x : s) {
can.insert(reach[x]);
}
}
{
int d = c;
for (auto [a, t] : qry1[d]) {
if (ans[t]) {
continue;
}
if (a == d) {
ans[t] = -1;
continue;
}
ans[t] = -1 + -(reach[a] >= t);
}
for (int t : qry2[d]) {
ans[t] += can.order_of_key(t) + 1;
}
}
alive[c] = false;
for (auto [u, _] : adj[c]) {
if (alive[u]) {
solve(u);
}
}
}
};
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
Solver t(n, k);
for (int i = 0; i < n - 1 + k; ++i) {
char op;
cin >> op;
if (op == 'S') {
int a, b;
cin >> a >> b;
--a, --b;
t.add_edge(a, b, i);
} else if (op == 'Q') {
int a, d;
cin >> a >> d;
--a, --d;
t.add_query1(a, d, i);
} else {
int d;
cin >> d;
--d;
t.add_query2(d, i);
}
}
t.solve();
for (int i = 0; i < n - 1 + k; ++i) {
int res = t.ans[i];
if (res) {
if (res > 0) {
cout << res << "\n";
} else {
cout << (res == -1 ? "yes" : "no") << "\n";
}
}
}
return 0;
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
22 ms |
3636 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
22 ms |
3636 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3408 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3408 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3420 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3420 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
20 ms |
3360 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
20 ms |
3360 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3416 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
18 ms |
3416 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
21 ms |
3428 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
21 ms |
3428 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
