【题解】JZOJ6703 tree

题意

给定 n n n 个点的树,每个点有点权,每次询问点权大于等于 x x x 的点构成的子图有多少个连通块,动态修改点权,保证修改后点权不小于修改前。

题解

首先有一个显然的结论:森林的连通块个数 = 点数 - 边数。

这就是说,对于一个询问 x x x,我们只需要知道大于等于 x x x 的点有多少个,所连两点均大于等于 x x x 的边有多少条,就能算出答案。

考虑动态维护大于等于 x x x 的点,发现这是一个单点修改区间查询,用一个统计后缀和的树状数组即可,每次修改就在原来的权值处减 1 1 1,在新的权值处加 1 1 1。查询 [ x , W m a x ] [x,W_{max}] [x,Wmax] 的和即可。

考虑怎么维护边。同样用一个树状数组统计所连接两点大于等于 x x x 的边。一条边的贡献显然是 min ⁡ ( a u , a v ) \min(a_u,a_v) min(au,av)

考虑 sub3,也就是菊花图。显然叶子节点的修改直接在树状数组上修改即可。对于根节点 1 1 1,开一个 multiset 维护大于等于 a 1 a_1 a1 的邻接点。对于 multiset 里面的点,其贡献显然是 a 1 a_1 a1,其它的点贡献就是 a x a_x ax。时间复杂度 O ( n log ⁡ n ) O(n\log n) O(nlogn)

考虑将这个做法拓展至每个点。每个点开 multiset s x s_x sx,存它的儿子里大于等于它的点。

对于一个修改 a x ← y a_x\larr y axy,考虑其儿子 s o n son son

  • a s o n < a x a_{son}<a_x ason<ax s o n son son 修改前后都不在 s x s_x sx 中,边 ( x , s o n ) (x,son) (x,son) 的贡献一直是 a s o n a_{son} ason
  • a s o n ≥ a x , a s o n < y a_{son}\ge a_x,a_{son}<y asonax,ason<y s o n son son 原本在 s x s_x sx 中,修改后不在,边 ( x , s o n ) (x,son) (x,son) 的贡献由 a x a_x ax 变为 a s o n a_{son} ason
  • a s o n ≥ y a_{son}\ge y asony s o n son son 修改前后都在 s x s_x sx 中,边 ( x , s o n ) (x,son) (x,son) 的贡献由 a x a_x ax 变为 y y y

考虑其父亲 f a fa fa

  • y < a f a y<a_{fa} y<afa x x x 修改前后都不在 s f a s_{fa} sfa 中,边 ( f a , x ) (fa,x) (fa,x) 的贡献由 a x a_x ax 变为 y y y
  • a x < a f a , y ≥ a f a a_x<a_{fa},y\ge a_{fa} ax<afa,yafa x x x 修改前不在 s f a s_{fa} sfa 中,修改后在 s f a s_{fa} sfa 中,边 ( f a , x ) (fa,x) (fa,x) 的贡献由 a x a_x ax 变为 a f a a_{fa} afa
  • a x ≥ a f a a_x\ge a_{fa} axafa x x x 修改前后都在 s f a s_{fa} sfa 中,边 ( f a , x ) (fa,x) (fa,x) 的贡献一直是 a f a a_{fa} afa

按照以上维护 s x s_x sx 和树状数组即可。

容易发现,所有节点的 multiset 的大小总和一定是递减的。将一个点 x x x 提升 d d d 后,若想让 x x x 再次加入 multiset,需要把 [ a x , a x + d ] [a_x,a_x+d] [ax,ax+d] 的点全部修改才能做到。所以这个修改均摊 O ( 1 ) O(1) O(1)。总时间复杂度 O ( n log ⁡ n ) O(n\log n) O(nlogn)

实现

这个树状数组的写法比较奇妙,是统计后缀和的。

注意不要漏了一些修改。

#include <bits/stdc++.h>
using namespace std;
#define fre(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
template<typename Ty> void read(Ty &x) {
	int c = getchar(), f = 1;
	for (; c < '0' || c > '9'; c = getchar()) if (c == '-') f = -1;
	for (x = 0; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
	x *= f;
}
#define lowbit(x) x & (-x)
const int N = 500005, W = 1000005;
int n, q, a[N], cnt = 0, fir[N], nxt[N << 1], to[N << 1], fa[N], tr[W], trp[W];
multiset<int> s[N];
void ade(int u, int v) {
	cnt++, nxt[cnt] = fir[u], fir[u] = cnt, to[cnt] = v;
	cnt++, nxt[cnt] = fir[v], fir[v] = cnt, to[cnt] = u;
}
void upd(int x, int val) { for (; x; x -= lowbit(x)) tr[x] += val; }
int gsum(int x) {
	int sum = 0;
	for (; x < W; x += lowbit(x)) sum += tr[x];
	return sum;
}
void updp(int x, int val) { for (; x; x -= lowbit(x)) trp[x] += val; }
int gsump(int x) {
	int sum = 0;
	for (; x < W; x += lowbit(x)) sum += trp[x];
	return sum;
}
void dfs(int r) {
	for (int i = fir[r]; i; i = nxt[i])
		if (to[i] != fa[r]) {
			fa[to[i]] = r, dfs(to[i]);
			if (a[to[i]] >= a[r]) s[r].insert(a[to[i]]);
			else upd(a[to[i]], 1);
		}
}
int main() {
	fre(tree);
	read(n), read(q);
	for (int i = 1; i <= n; i++) read(a[i]), updp(a[i], 1);
	for (int i = 1, u, v; i < n; i++) read(u), read(v), ade(u, v);
	dfs(1);
	for (int i = 1; i <= n; i++)
		if (!s[i].empty())
			upd(a[i], s[i].size());
	for (int t = 1, opt, x, y; t <= q; t++) {
		read(opt);
		if (opt == 1) {
			read(x), read(y);
			if (fa[x]) {
				if (a[x] >= a[fa[x]]) s[fa[x]].erase(s[fa[x]].find(a[x]));
				else upd(a[x], -1), upd(min(y, a[fa[x]]), 1);
				if (y >= a[fa[x]]) s[fa[x]].insert(y);
			}
			int cn = 0;
			while (!s[x].empty() && *s[x].begin() < y) upd(*s[x].begin(), 1), s[x].erase(s[x].begin()), cn++;
			upd(a[x], -cn), upd(a[x], -s[x].size()), upd(y, s[x].size()), updp(a[x], -1), updp(a[x] = y, 1);
		}
		else read(x), printf("%d\n", gsump(x) - gsum(x));
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值