【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;
}
【SPOJ】QTREE6-Query on a tree VI的更多相关文章
- 【SPOJ】375. Query on a tree(树链剖分)
http://www.spoj.com/problems/QTREE/ 这是按边分类的. 调试调到吐,对拍都查不出来,后来改了下造数据的,拍出来了.囧啊啊啊啊啊啊 时间都花在调试上了,打hld只用了半 ...
- SPOJ 16549 - QTREE6 - Query on a tree VI 「一种维护树上颜色连通块的操作」
题意 有操作 $0$ $u$:询问有多少个节点 $v$ 满足路径 $u$ 到 $v$ 上所有节点(包括)都拥有相同的颜色$1$ $u$:翻转 $u$ 的颜色 题解 直接用一个 $LCT$ 去暴力删边连 ...
- 【SPOJ】QTREE6(Link-Cut-Tree)
[SPOJ]QTREE6(Link-Cut-Tree) 题面 Vjudge 题解 很神奇的一道题目 我们发现点有黑白两种,又是动态加边/删边 不难想到\(LCT\) 最爆力的做法,显然是每次修改单点颜 ...
- 【BZOJ1803】Spoj1487 Query on a tree III 主席树+DFS序
[BZOJ1803]Spoj1487 Query on a tree III Description You are given a node-labeled rooted tree with n n ...
- QTREE6 - Query on a tree VI 解题报告
QTREE6 - Query on a tree VI 题目描述 给你一棵\(n\)个点的树,编号\(1\)~\(n\).每个点可以是黑色,可以是白色.初始时所有点都是黑色.下面有两种操作请你操作给我 ...
- 【SPOJ】10628. Count on a tree(lca+主席树+dfs序)
http://www.spoj.com/problems/COT/ (速度很快,排到了rank6) 这题让我明白了人生T_T 我知道我为什么那么sb了. 调试一早上都在想人生. 唉. 太弱. 太弱. ...
- 【bzoj1803】Spoj1487 Query on a tree III DFS序+主席树
题目描述 You are given a node-labeled rooted tree with n nodes. Define the query (x, k): Find the node w ...
- 树链剖分【p4116】Qtree3 - Query on a tree
Description 给出N个点的一棵树(N-1条边),节点有白有黑,初始全为白 有两种操作: 0 i : 改变某点的颜色(原来是黑的变白,原来是白的变黑) 1 v : 询问1到v的路径上的第一个黑 ...
- LCA【SP913】Qtree - Query on a tree II
Description 给定一棵n个点的树,边具有边权.要求作以下操作: DIST a b 询问点a至点b路径上的边权之和 KTH a b k 询问点a至点b有向路径上的第k个点的编号 有多组测试数据 ...
- SPOJ QTREE6 Query on a tree VI 树链剖分
题意: 给出一棵含有\(n(1 \leq n \leq 10^5)\)个节点的树,每个顶点只有两种颜色:黑色和白色. 一开始所有的点都是黑色,下面有两种共\(m(1 \leq n \leq 10^5) ...
随机推荐
- 剖析Hadoop和Spark的Shuffle过程差异(一)
一.前言 对于基于MapReduce编程范式的分布式计算来说,本质上而言,就是在计算数据的交.并.差.聚合.排序等过程.而分布式计算分而治之的思想,让每个节点只计算部分数据,也就是只处理一个分片,那么 ...
- Github Desktop 克隆仓库一直出现“Authentication failed”
如图所示: 解决方法: 使用ssh链接下载而不是https
- redirect和forward的区别
异同 1.从地址栏显示来说 forward是服务器请求资源,服务器直接访问目标地址的URL,把那个URL的响应内容读取过来,然后把这些内容再发给浏览器.浏览器根本不知道服务器发送的内容从哪里来的,所以 ...
- 洛谷_Cx的故事_解题报告_第四题70
1.并查集求最小生成树 Code: #include <stdio.h> #include <stdlib.h> struct node { long x,y,c; ...
- 61. Rotate List(M);19. Remove Nth Node From End of List(M)
61. Rotate List(M) Given a list, rotate the list to the right by k places, where k is non-negative. ...
- linux c 编程 ------ 串口编程
http://blog.csdn.net/specialshoot/article/details/50707965 对于串口的打开操作,必须使用O_NOCTTY参数.O_NOCTTY如果路径名指向终 ...
- C#生成和识别二维码
用到外部一个DLL文件(ThoughtWorks.QRCode.dll),看效果 生成截图 识别截图 生成二维码后右键菜单可以保存二维码图片,然后可以到识别模式下进行识别,当然生成后可以用手机扫描识别 ...
- webpack进阶--loader
webpack的核心就是它的配置文件,只要配置好配置文件webpack就可以用得利索-- 而配置文件主要就是7个部分entry.output.plugins.resolve.devserver(web ...
- bzoj千题计划239:bzoj4069: [Apio2015]巴厘岛的雕塑
http://www.lydsy.com/JudgeOnline/problem.php?id=4069 a!=1: 从高位到低位一位一位的算 记录下哪些位必须为0 dp[i][j] 表示前i个数分为 ...
- .NET面试题系列(九)C# 结构体与类的区别
谈一下什么时候使用结构,什么使用类. 我们知道,结构存储在栈中,而栈有1个特点,就是空间较小,但是访问速度较快,堆空间较大,但是访问速度相对较慢.所以当我们描述1个轻量级对象的时候,可以将其定义为结构 ...