トピックリンク:https://www.luogu.org/problem/P2146
この質問はアルゴリズムが含まれます。
- ツリーチェーン分割;
- セグメントツリー(更新間隔と怠惰なマークを含む合計)
次にそれぞれのinstall x
、次のものが必要x
に1
ルート1に設定されているすべての上記のポイントを。
そして、1 num1をセットする前に、ノードの数に関する統計は、
統計は、ノードのnum2の番号の後に1に設定見て、
答えがあるnum2 - num1
(もちろん、あなたがすることができ、ノードの深さdep[x]
のノードの数を取得します)。
それぞれについてunistall x
、我々はする必要がありx
、すべてが上記の点0をルートとするサブツリーに設定します。
そして、NUM1 1の右側の値のノード数についての統計の前に0に設定され、
統計が1の右からnum2値内のノード数の後に0に設定見て、
答えがあるnum1-num2
(もちろん、num2
実際には、同等です0
)。
次のようにコードは次のとおりです。
#include <bits/stdc++.h>
using namespace std;
#define INF (1<<29)
const int maxn = 100010;
int fa[maxn],
dep[maxn],
size[maxn],
son[maxn],
top[maxn],
seg[maxn], seg_cnt,
rev[maxn],
n,
sumv[maxn<<2], lazy[maxn<<2];
vector<int> g[maxn];
void dfs1(int u, int p) {
size[u] = 1;
for (vector<int>::iterator it = g[u].begin(); it != g[u].end(); it ++) {
int v = (*it);
if (v == p) continue;
fa[v] = u;
dep[v] = dep[u] + 1;
dfs1(v, u);
size[u] += size[v];
if (size[v] >size[son[u]]) son[u] = v;
}
}
void dfs2(int u, int tp) {
seg[u] = ++seg_cnt;
rev[seg_cnt] = u;
top[u] = tp;
if (son[u]) dfs2(son[u], tp);
for (vector<int>::iterator it = g[u].begin(); it != g[u].end(); it ++) {
int v = (*it);
if (v == fa[u] || v == son[u]) continue;
dfs2(v, v);
}
}
#define lson l, mid, rt<<1
#define rson mid+1, r, rt<<1|1
void push_down(int rt, int len) {
if (lazy[rt] != -1) {
int l_len=len-len/2, r_len = len/2;
lazy[rt<<1] = lazy[rt];
lazy[rt<<1|1] = lazy[rt];
sumv[rt<<1] = lazy[rt] * l_len;
sumv[rt<<1|1] = lazy[rt] * r_len;
lazy[rt] = -1;
}
}
void push_up(int rt) {
sumv[rt] = sumv[rt<<1] + sumv[rt<<1|1];
}
void build(int l, int r, int rt) {
lazy[rt] = -1;
int mid = (l + r) / 2;
if (l == r) {
sumv[rt] = 0;
return;
}
build(lson); build(rson);
push_up(rt);
}
void update(int L, int R, long long v, int l, int r, int rt) {
if (L <= l && r <= R) {
sumv[rt] = (r-l+1) * v;
lazy[rt] = v;
return;
}
push_down(rt, r-l+1);
int mid = (l + r) / 2;
if (L <= mid) update(L, R, v, lson);
if (R > mid) update(L, R, v, rson);
push_up(rt);
}
long long query_sum(int L, int R, int l, int r, int rt) {
if (L <= l && r <= R) return sumv[rt];
push_down(rt, r-l+1);
int mid = (l + r) / 2;
long long tmp = 0;
if (L <= mid) tmp += query_sum(L, R, lson);
if (R > mid) tmp += query_sum(L, R, rson);
return tmp;
}
int t_ask(int u) {
int res = 0;
while (top[u] != 1) {
res += query_sum(seg[top[u]], seg[u], 1, n, 1);
u = fa[top[u]];
}
res += query_sum(seg[1], seg[u], 1, n, 1);
return res;
}
void t_update(int u) {
while (top[u] != 1) {
update(seg[top[u]], seg[u], 1, 1, n, 1);
u = fa[top[u]];
}
update(seg[1], seg[u], 1, 1, n, 1);
}
int m, x;
string op;
int main() {
cin >> n;
for (int i = 2; i <= n; i ++) {
cin >> x;
g[x+1].push_back(i);
}
dep[1] = fa[1] = 1;
dfs1(1, -1);
dfs2(1, 1);
build(1, n, 1);
cin >> m;
while (m --) {
cin >> op >> x;
x ++;
if (op == "install") {
int num1 = t_ask(x);
t_update(x);
int num2 = t_ask(x);
cout << num2 - num1 << endl;
}
else { // uninstall
int num1 = query_sum(seg[x], seg[x]+size[x]-1, 1, n, 1);
update(seg[x], seg[x]+size[x]-1, 0, 1, n, 1);
int num2 = query_sum(seg[x], seg[x]+size[x]-1, 1, n, 1);
cout << num1 - num2 << endl;
}
}
return 0;
}
著者:zifeiy