洛谷P2146 树链剖分
思路:直接树链剖分,用线段树维护即可,算是树剖的经典题目吧。
代码:
#include <bits/stdc++.h>
#define ls(x) (x << 1)
#define rs(x) ((x << 1) | 1)
using namespace std;
const int maxn = 100010;
int head[maxn], Next[maxn * 2], ver[maxn * 2];
int sz[maxn], son[maxn], d[maxn], dfn[maxn], top[maxn], f[maxn];
int tot, cnt;
int n;
struct SegmentTree {
int val, lz;
int l, r;
};
SegmentTree tr[maxn * 4];
void add(int x, int y) {
ver[++tot] = y;
Next[tot] = head[x];
head[x] = tot;
}
void dfs1(int x, int fa = -1) {
sz[x] = 1;
f[x] = fa;
int mx = 0;
for (int i = head[x]; i; i = Next[i]) {
int y = ver[i];
if(y == fa) continue;
d[y] = d[x] + 1;
dfs1(y, x);
sz[x] += sz[y];
if(sz[y] > mx) {
mx = sz[y];
son[x] = y;
}
}
}
void dfs2(int x, int fa, int t) {
dfn[x] = ++cnt;
top[x] = t;
if(son[x]) dfs2(son[x], x, t);
for (int i = head[x]; i; i = Next[i]) {
int y = ver[i];
if(y == fa || y == son[x]) continue;
dfs2(y, x, y);
}
}
void pushup(int o) {
tr[o].val = tr[ls(o)].val + tr[rs(o)].val;
}
void maintain(int o, int val) {
tr[o].val = val * (tr[o].r - tr[o].l + 1);
tr[o].lz = val;
}
void pushdown(int o) {
if(tr[o].lz != -1) {
maintain(ls(o), tr[o].lz);
maintain(rs(o), tr[o].lz);
tr[o].lz = -1;
}
}
void build(int o, int l, int r) {
tr[o].l = l, tr[o].r = r;
if(l == r) {
tr[o].val = 0;
tr[o].lz = -1;
return;
}
int mid = (l + r) >> 1;
build(ls(o), l, mid);
build(rs(o), mid + 1, r);
pushup(o);
}
void update(int o, int l, int r, int ql, int qr, int val) {
if(l >= ql && r <= qr) {
tr[o].val = (r - l + 1) * val;
tr[o].lz = val;
return;
}
pushdown(o);
int mid = (l + r) >> 1;
if(ql <= mid) update(ls(o), l, mid, ql, qr, val);
if(qr > mid) update(rs(o), mid + 1, r, ql, qr, val);
pushup(o);
}
int query(int o, int l, int r, int ql, int qr) {
if(l >= ql && r <= qr) {
return tr[o].val;
}
pushdown(o);
int mid = (l + r) >> 1, ans = 0;
if(ql <= mid) ans += query(ls(o), l, mid , ql, qr);
if(qr > mid) ans += query(rs(o), mid + 1, r, ql, qr);
return ans;
}
int solve(int x) {
int ans = 0, st = x;
while(x != -1) {
ans += query(1, 1, n, dfn[top[x]], dfn[x]);
x = f[top[x]];
}
return d[st] - d[0] + 1 - ans;
}
void update1(int x, int val) {
while(x != -1) {
update(1, 1, n, dfn[top[x]], dfn[x], val);
x = f[top[x]];
}
}
char s[110];
int main() {
int x, m;
scanf("%d", &n);
for (int i = 1; i < n; i++) {
scanf("%d", &x);
add(x, i);
add(i, x);
}
f[0] = -1;
build(1, 1, n);
dfs1(0);
dfs2(0, -1, 0);
scanf("%d", &m);
while(m--) {
scanf("%s", s + 1);
if(s[1] == 'i') {
scanf("%d", &x);
int tmp = query(1, 1, n, dfn[x], dfn[x]);
if(tmp == 1) {
printf("0\n");
continue;
}
printf("%d\n", solve(x));
update1(x, 1);
} else {
scanf("%d", &x);
int tmp = query(1, 1, n, dfn[x], dfn[x]);
if(tmp == 0) {
printf("0\n");
continue;
}
printf("%d\n", query(1, 1, n, dfn[x], dfn[x] + sz[x] - 1));
update(1, 1, n, dfn[x], dfn[x] + sz[x] - 1, 0);
}
}
}
洛谷P2146 树链剖分的更多相关文章
- 【算法学习】【洛谷】树链剖分 & P3384 【模板】树链剖分 P2146 软件包管理器
刚学的好玩算法,AC2题,非常开心. 其实很早就有教过,以前以为很难就没有学,现在发现其实很简单也很有用. 更重要的是我很好调试,两题都是几乎一遍过的. 介绍树链剖分前,先确保已经学会以下基本技巧: ...
- 洛谷P3384 树链剖分
如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式: 2 x ...
- 洛谷 P3384 树链剖分(模板题)
题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式 ...
- 【树链剖分】洛谷P3379 树链剖分求LCA
题目描述 如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先. 输入输出格式 输入格式: 第一行包含三个正整数N.M.S,分别表示树的结点个数.询问的个数和树根结点的序号. 接下来N-1行每 ...
- 洛谷 P3384树链剖分 题解
题面 挺好的一道树剖模板: 首先要学会最模板的树剖: 然后这道题要注意几个细节: 初始化时,seg[0]=1,seg[root]=1,top[root]=root,rev[1]=root; 在线段树上 ...
- 洛谷 [P3384] 树链剖分 模版
支持各种数据结构上树,注意取膜. #include <iostream> #include <cstring> #include <algorithm> #incl ...
- 树链剖分详解(洛谷模板 P3384)
洛谷·[模板]树链剖分 写在前面 首先,在学树链剖分之前最好先把 LCA.树形DP.DFS序 这三个知识点学了 emm还有必备的 链式前向星.线段树 也要先学了. 如果这三个知识点没掌握好的话,树链剖 ...
- 【模板时间】◆模板·II◆ 树链剖分
[模板·II]树链剖分 学长给我讲树链剖分,然而我并没有听懂,还是自学有用……另外感谢一篇Blog +by 自为风月马前卒+ 一.算法简述 树链剖分可以将一棵普通的多叉树转为线段树计算,不但可以实现对 ...
- 洛谷P2146 [NOI2015]软件包管理器 题解 树链剖分+线段树
题目链接:https://www.luogu.org/problem/P2146 本题涉及算法: 树链剖分: 线段树(区间更新及求和,涉及懒惰标记) 然后对于每次 install x ,需要将 x 到 ...
随机推荐
- nodeJs爬虫小程序练习
//爬虫小程序 var express = require('express'); //superagent是一个http的库,可以发起get和post请求 var superagent = requ ...
- python中的单引号双引号和三引号
python的单引号和双引号没有本质的区别,而三引号有两种作用:注释和换行 一. 单引号中可以包含双引号,双引号中可以包含单引号 print("好好学习,'天天向上'") 结果: ...
- 缓存淘汰算法--LRU算法(转)
(转自:http://flychao88.iteye.com/blog/1977653) 1. LRU1.1. 原理 LRU(Least recently used,最近最少使用)算法根据数据的历史访 ...
- Shiro安全配置
主要还是整合了本地ehcache,集群session管理过段时间放出 <?xml version="1.0" encoding="UTF-8"?> ...
- 红帽企业版Linux成为Linux下的.NET Core的参考平台
微软和红帽声明将在红帽企业版Linux运行的.NET纳入官方支持.经两家公司透露,“红帽企业级Linux将成为Linux下的.NET Core主要参考操作系统”. 来自红帽资深开发者Harry Mow ...
- 如何下载中文语言包让Eclipse汉化?
Eclipse软件是JAVA开发必备的软件,对于英语不好的人而言面对英文代码还好,但是软件也是英语的那就头疼了,怎么解决?可以去官网下载中文语言包解压到Eclipse安装目录就可以汉化了. 工具/ ...
- 三目运算符与if else的运行效率
两者的效率比较: 当比较次数较少时,效率一样: 当比较次数较多时,发现ifelse的速度更快,应该是其汇编指令更少的原因. if else的汇编代码如下: 三目运算符代码如下:
- Arc083_F Collecting Balls
传送门 题目大意 给定$N$,在$(1,0),(2,0)......(N,0)$和$(0,1),(0,2)...(0,N)$上都有$1$个机器人,同时给定$2N$个坐标$(x,y),x,y\in[1, ...
- Excel分类汇总
版本:2016,数据来源:我要自学网,曾贤志老师 1.首先,要进行分类,在进行汇总, 如对日期进行汇总的话: 如果对品名进行汇总的话 2.光标要定在数据源,在选择分类字段(包含项目比如品名 ...
- 用PowerShell在China Azure创建ARM虚拟机
Azure目前有两种工作模式:ASM和ARM. 在国内的Azure,我们都是使用ASM的模式.但这种模式有很多限制,比如每个VM必须有一个公网地址,部署不能批量部署等等.ARM对Azure的整体架构做 ...