【SPOJ】QTREE6-Query on a tree VI
题解
老年选手的代码康复计划QAQ
这题又没一遍A,难受
每个节点维护这个节点子树内联通块的大小
维护所有节点轻儿子的\(g[u][0]\)表示所有轻儿子白色的联通块总数
\(g[u][1]\)表示所有轻儿子黑色联通块总数
更新一个点为新颜色的时候,是\(g[u][c[u] ^ 1] + 1\)再加上重儿子价值(如果修改后颜色相同
然后我们向上更新,修改到和父亲节点颜色相同的最浅的节点
如果改后颜色和父亲颜色不同,那么这些点都减去原来的值
否则都加上新值
然后在经过轻边的时候顺带维护一下\(g\)即可
对于找颜色相同的最浅节点,用线段树维护一个区间颜色是否相同即可
代码
#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define pdi pair<db,int>
#define mp make_pair
#define pb push_back
#define enter putchar('\n')
#define space putchar(' ')
#define eps 1e-8
#define mo 974711
#define MAXN 100005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef double db;
template<class T>
void read(T &res) {
res = 0;char c = getchar();T f = 1;
while(c < '0' || c > '9') {
if(c == '-') f = -1;
c = getchar();
}
while(c >= '0' && c <= '9') {
res = res * 10 + c - '0';
c = getchar();
}
res *= f;
}
template<class T>
void out(T x) {
if(x < 0) {x = -x;putchar('-');}
if(x >= 10) {
out(x / 10);
}
putchar('0' + x % 10);
}
int N,M;
struct node {
int to,next;
}E[MAXN * 2];
int sumE,head[MAXN],c[MAXN];
int dfn[MAXN],son[MAXN],siz[MAXN],top[MAXN],dep[MAXN],fa[MAXN],idx,seq[MAXN];
int g[MAXN][2];
void add(int u,int v) {
E[++sumE].to = v;
E[sumE].next = head[u];
head[u] = sumE;
}
void dfs1(int u) {
dep[u] = dep[fa[u]] + 1;
siz[u] = 1;
for(int i = head[u] ; i ; i = E[i].next) {
int v = E[i].to;
if(v != fa[u]) {
fa[v] = u;
dfs1(v);
siz[u] += siz[v];
if(siz[v] > siz[son[u]]) son[u] = v;
}
}
for(int i = head[u] ; i ; i = E[i].next) {
int v = E[i].to;
if(v != fa[u] && v != son[u]) {
g[u][1] += siz[v];
}
}
}
void dfs2(int u) {
dfn[u] = ++idx;seq[idx] = u;
if(!top[u]) top[u] = u;
if(son[u]) {top[son[u]] = top[u];dfs2(son[u]);}
for(int i = head[u] ; i ; i = E[i].next) {
int v = E[i].to;
if(v != fa[u] && v != son[u]) dfs2(v);
}
}
namespace seg_tr {
struct node {
int L,R;
int val,lz;
bool sc,col;
}tr[MAXN * 4];
#define lc(u) u << 1
#define rc(u) u << 1 | 1
#define val(u) tr[u].val
#define lz(u) tr[u].lz
#define sc(u) tr[u].sc
#define col(u) tr[u].col
void addlz(int u,int v) {
val(u) += v;lz(u) += v;
}
void push_down(int u) {
if(tr[u].lz) {
addlz(lc(u),tr[u].lz);
addlz(rc(u),tr[u].lz);
}
tr[u].lz = 0;
}
void update(int u) {
if(sc(lc(u)) && sc(rc(u)) && !(col(lc(u)) ^ col(rc(u)) ) ) {
sc(u) = 1;col(u) = col(lc(u));
}
else sc(u) = 0;
}
void build(int u,int l,int r) {
tr[u].L = l;tr[u].R = r;
if(l == r) {
tr[u].sc = 1;tr[u].col = c[seq[l]];
tr[u].val = siz[seq[l]];tr[u].lz = 0;
return;
}
int mid = (l + r) >> 1;
build(u << 1,l,mid);
build(u << 1 | 1,mid + 1,r);
update(u);
}
int query(int u,int p) {
if(tr[u].L == tr[u].R) return tr[u].val;
push_down(u);
int mid = (tr[u].L + tr[u].R) >> 1;
if(p <= mid) return query(lc(u),p);
else if(p > mid) return query(rc(u),p);
}
void add(int u,int l,int r,int v) {
if(tr[u].L == l && tr[u].R == r) {addlz(u,v);return;}
push_down(u);
int mid = (tr[u].L + tr[u].R) >> 1;
if(r <= mid) add(lc(u),l,r,v);
else if(l > mid) add(rc(u),l,r,v);
else {
add(lc(u),l,mid,v);
add(rc(u),mid + 1,r,v);
}
}
void change(int u,int p,int v,bool on) {
if(tr[u].L == tr[u].R) {
tr[u].val = v;
tr[u].col = on;
return;
}
push_down(u);
int mid = (tr[u].L + tr[u].R) >> 1;
if(p <= mid) change(lc(u),p,v,on);
else change(rc(u),p,v,on);
update(u);
}
int query_sc(int u,int l,int r) {
if(tr[u].L == l && tr[u].R == r) return tr[u].sc;
push_down(u);
int mid = (tr[u].L + tr[u].R) >> 1;
if(r <= mid) return query_sc(lc(u),l,r);
else if(l > mid) return query_sc(rc(u),l,r);
else {
return query_sc(lc(u),l,mid) && query_sc(rc(u),mid + 1,r) && c[seq[mid]] == c[seq[mid + 1]];
}
}
int Query(int p) {
return query(1,dfn[p]);
}
void Add(int l,int r,int v) {
add(1,l,r,v);
}
void Change(int p,int v,bool on) {
change(1,dfn[p],v,on);
}
bool Query_sc(int l,int r) {
return query_sc(1,l,r);
}
}
int Find_same_color(int u) {
while(1) {
if(seg_tr::Query_sc(dfn[top[u]],dfn[u])) {
if(top[u] == 1) return 1;
if(c[fa[top[u]]] != c[u]) return top[u];
u = fa[top[u]];
}
else {
int L = dfn[top[u]] + 1,R = dfn[u];
while(L < R) {
int mid = (L + R) >> 1;
if(seg_tr::Query_sc(mid,dfn[u])) R = mid;
else L = mid + 1;
}
return seq[L];
}
}
}
void Change_Path(int u,int v,int d) {
while(top[u] != top[v]) {
seg_tr::Add(dfn[top[u]],dfn[u],d);
g[fa[top[u]]][c[top[u]]] += d;
u = fa[top[u]];
}
seg_tr::Add(dfn[v],dfn[u],d);
if(v == top[u]) g[fa[v]][c[v]] += d;
}
void Init() {
read(N);
int u,v;
for(int i = 1 ; i <= N ; ++i) c[i] = 1;
for(int i = 1 ; i < N ; ++i) {
read(u);read(v);
add(u,v);add(v,u);
}
dfs1(1);dfs2(1);
seg_tr::build(1,1,N);
}
void Solve() {
read(M);
int op,u;
for(int i = 1 ; i <= M ; ++i) {
read(op);read(u);
if(!op) {
out(seg_tr::Query(Find_same_color(u)));enter;
}
else {
int nv = 1 + g[u][c[u] ^ 1] + (son[u] && c[son[u]] == (c[u] ^ 1) ? seg_tr::Query(son[u]) : 0);
int ov = seg_tr::Query(u);
if(u == 1) {
seg_tr::Change(u,nv,c[u] ^ 1);c[u] ^= 1;
}
else {
if(u == top[u]) {
g[fa[u]][c[u]] -= ov;
g[fa[u]][c[u] ^ 1] += nv;
}
int t = Find_same_color(fa[u]);
if(c[fa[u]] == c[u]) Change_Path(fa[u],t,-ov);
else Change_Path(fa[u],t,nv);
seg_tr::Change(u,nv,c[u] ^ 1);c[u] ^= 1;
}
}
}
}
int main() {
#ifdef ivorysi
freopen("f1.in","r",stdin);
#endif
Init();
Solve();
return 0;
}