bzoj 3730. 震波(动态点分治 + vector树状数组)_bzoj 震波-程序员宅基地

技术标签: 树状数组  vector开树状数组  动态点分治  

在这里插入图片描述


(自己写的动态点分治巨垃圾,常数是别人的两倍)
用动态开点线段树死活过不去,学了一波大佬用 vector 开树状数组立马就卡过去了

考虑点分树的做法,在点分树上每个点以距离为下标建一棵线段树,每次询问查询子树的贡献,再暴力向上跳合并父节点来自其它节点的贡献。

因为子树树形被破坏,在做减法时,子节点子树对父节点的贡献不能用子树维护的信息 + 连向父节点的边的贡献得到,考虑在每个节点再维护一个线段树,按距离建树用来统计子树内的节点到其父节点距离为 p 的点权和,这样便可以暴力爬树统计子树外的节点的贡献。

理论复杂度为 O ( n log ⁡ 2 n ) O(n \log^2 n) O(nlog2n),但常数巨大,况且还要上动态开点线段树。

采用了别人的用vector开树状数组的做法,首先对空间进行分析:
动态开点线段树:由于树高 l o g log log,每次维护要暴力向上爬维护路径上所有节点的信息,显然一个点最多在线段树插入 log ⁡ \log log 次,每次插入是单点修改,最多需要新建 log ⁡ \log log 个节点,理论空间复杂度上限是 n log ⁡ 2 n n \log^2 n nlog2n

v e c t o r vector vector 开树状数组: v e c t o r vector vector 开树状数组最主要要掌握每个节点要开的大小,由于我们维护的是子树信息并且以距离为关键字进行建树维护。每个节点的树状数组大小 = 该节点在点分树上子节点的个数。由于点分树每个节点都是这棵子树的重心,每个子节点的 s i z e size size 的大小大约为父节点 s i z e size size 1 2 \frac{1}{2} 21,总空间大约为 n + 1 2 n + 1 2 n + 1 4 n + . . . + 1 n + \frac{1}{2} n + \frac{1}{2}n + \frac{1}{4}n + ... + 1 n+21n+21n+41n+...+1,大概是 n log ⁡ n n\log n nlogn 级别(不是很准确,大概是这个数字左右)

v e c t o r vector vector 调用 r e s i z e ( s i z e ) resize(size) resize(size) 函数就可以分配, s i z e size size 大小的空间,如果再传一个参数,将全部赋值为第二个参数的值,最后用树状数组比用线段树快一倍以上。


代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
typedef long long ll;
#define lowbit(i) (i & (-i))
struct Graph {
    
	int head[maxn], to[maxn << 1], cnt, nxt[maxn << 1], w[maxn << 1];
	void init() {
    
		memset(head,-1,sizeof head);
		cnt = 0;
	}
	void add(int u,int v,int c) {
    
		to[cnt] = v;
		nxt[cnt] = head[u];
		w[cnt] = c;
		head[u] = cnt++;
		
		to[cnt] = u;
		nxt[cnt] = head[v];
		w[cnt] = c;
		head[v] = cnt++;
	}
}G;
vector<int> sum[maxn],fsum[maxn];					//动态开点树状数组 
void modify(vector<int> &tree,int p,int v) {
    
	if (p == 0)
		tree[0] += v;
	else {
    
		for (int i = p; i < tree.size(); i += lowbit(i))
			tree[i] += v;	
	}
}
int ask(vector<int> &tree,int k) {
    
	k = min(k,(int) tree.size() - 1);
	int ans = tree[0];
	for (int i = k; i; i -= lowbit(i))
		ans += tree[i];
	return ans;
}
int st[maxn][30],d[maxn],cnt,a[maxn];
int fir[maxn],vis[maxn],f[maxn],root,sz[maxn],siz,p[maxn];
int n,q,RT;
inline int read()
{
    
	int x=0,f=1;char ch;
	while(ch<'0'||ch>'9'){
    if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){
    x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void dfs1(int u,int fa) {
    			//预处理深度,st表,以及欧拉序 
	fir[u] = ++cnt; st[cnt][0] = u;
	for (int i = G.head[u]; i + 1; i = G.nxt[i]) {
    
		int v = G.to[i];
		if (v == fa) continue;
		d[v] = d[u] + G.w[i];
		dfs1(v,u);
		st[++cnt][0] = u;
	}
}
void dfs2(int u,int fa) {
    			//求一棵子树的重心 
	sz[u] = 1; f[u] = 0;
	for (int i = G.head[u]; i + 1; i = G.nxt[i]) {
    
		int v = G.to[i];
		if (v == fa || vis[v]) continue;
		dfs2(v,u);
		sz[u] += sz[v];
		if (sz[v] > f[u]) f[u] = sz[v];
	}
	if (siz - sz[u] > f[u]) f[u] = siz - sz[u];
	if (!root || f[u] < f[root]) root = u;
}
void dfs3(int u,int fa) {
     			//点分构建点分树 
	vis[u] = 1;
	int all = siz;
	sum[u].resize(all + 1), fsum[u].resize(all + 1);
	for (int i = G.head[u]; i + 1; i = G.nxt[i]) {
    
		int v = G.to[i];
		if (v == fa || vis[v]) continue;
		root = 0, siz = sz[v]; 
		if (siz > sz[u]) siz = all - sz[u];
		dfs2(v,u);
		p[root] = u;	//构建点分树
		dfs3(root,u); 
	}
}
int cal(int u,int v) {
    
	return d[u] < d[v] ? u : v;
}
int getlca(int x,int y) {
    
	if (fir[x] > fir[y]) swap(x,y);
	int p = log2(fir[y] - fir[x] + 1);
	return cal(st[fir[x]][p],st[fir[y] - (1 << p) + 1][p]);
}
int getdis(int x,int y) {
    
	return d[x] + d[y] - 2 * d[getlca(x,y)];
}
void prework() {
    
	dfs1(1,0);
	for (int i = 1; i <= log2(cnt); i++)
		for (int j = 1; j + (1 << i) - 1 <= cnt; j++)
			st[j][i] = cal(st[j][i - 1],st[j + (1 << (i - 1))][i - 1]);
	root = 0, siz = n; dfs2(1,0);
	RT = root; dfs3(root,0);		//保留第一次的根节点,并构建点分树 
}
void update(int x,int y) {
    
	for (int i = x; i; i = p[i]) {
    
		int dist = getdis(x,i);
		modify(sum[i],dist,y);
		if (p[i]) modify(fsum[i],getdis(x,p[i]),y);
	}
}
int qry(int x,int k) {
    
	int tmp = ask(sum[x],k);
	for (int i = x; p[i]; i = p[i]) {
    
		int dist = getdis(x,p[i]);
		if (dist <= k) {
    
			tmp += ask(sum[p[i]],k - dist);
			tmp -= ask(fsum[i],k - dist);	
		}
	}
	return tmp;
}
int main() {
    
	n = read(); q = read();
	G.init();
	for (int i = 1; i <= n; i++) {
    
		a[i] = read();
	}
	for (int i = 1; i < n; i++) {
    
		int u,v; u = read(); v = read();
		G.add(u,v,1);
	}
	prework();
	for (int i = 1; i <= n; i++)
		update(i,a[i]);
	int ans = 0;
	while (q--) {
    
		int op,x,y; op = read(); x = read(); y = read();
		x ^= ans; y ^= ans;
		if (op == 1) {
    
			update(x,y - a[x]);
			a[x] = y;
		} else {
    
			ans = qry(x,y);
			printf("%d\n",ans);	
		}
	}
	return 0;
}
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_41997978/article/details/104415608

智能推荐

王斌老师的博客_王斌 github-程序员宅基地

文章浏览阅读480次。http://blog.sina.com.cn/s/blog_736d0b9101018cgc.html_王斌 github

ACM OJ Collection_htt//acm.wydtang.top/-程序员宅基地

文章浏览阅读737次。原文来自:http://blog.csdn.net/hncqp/article/details/4455263 ACM OJ Collection(排名不分先后):中国:浙江大学(ZJU):http://acm.zju.edu.cn/北京大学(PKU):htt_htt//acm.wydtang.top/

ios 自己服务器 苹果支付_修复苹果IOS支付-程序员宅基地

文章浏览阅读467次。更新记录1.0.0(2019-07-01)插件简介专门用来修复苹果IOS支付时出现"您已购买此App内购买项目。此项目将免费恢复"。问题描述首先在IOS平台里面创建“APP内购买项目”,选择的是“消耗型项目”,然后用uni-app官方的支付api进行支付,多支付几次,有时候就会出现提示“您已购买此App内购买项目。此项目将免费恢复”,特别是在沙盒测试里面支付很大几率出现,我明明选的是消耗型项目,应..._ios开发苹果支付恢复权益

spring MVC mock类单元测试(controller)_mvcmock-程序员宅基地

文章浏览阅读5.6k次。Spring从J2EE的Web端为每个关键接口提供了一个mock实现:MockHttpServletRequest几乎每个单元测试中都要使用这个类,它是J2EE Web应用程序最常用的接口HttpServletRequest的mock实现。MockHttpServletResponse此对象用于HttpServletRespons_mvcmock

【我的世界Minecraft-MC】常见及各种指令大杂烩【2022.8版】_summon生成掉落物-程序员宅基地

文章浏览阅读8.5k次,点赞7次,收藏22次。execute as @a at @s run clear @s minecraft:dark_oak_planks{display:{Name:“{“text”:“第三关[阴森古堡]”,“color”:“red”,“italic”:false}”,color:“16711680”},Enchantments:[{id:“protection”,lvl:1}],Unbreakable:1b} 1。Lore:[“{“text”:“免费”,“color”:“blue”,“italic”:false}”]..._summon生成掉落物

CentOS 7安装教程(图文详解)_centos 安装-程序员宅基地

文章浏览阅读10w+次,点赞487次,收藏2.1k次。CentOS 7安装教程: 准备: 软件:VMware Workstation 镜像文件:CentOS-7-x86_64-bin-DVD1.iso (附:教程较为详细,注释较多,故将操作的选项进行了加粗字体显示。) 1、文件--新建虚拟机--自定义 2、..._centos 安装

随便推点

Github项目分享——免费的画图工具drow,前端插件化面试_draw github画图-程序员宅基地

文章浏览阅读333次,点赞3次,收藏3次。项目介绍一款很好用的免费画图软件,支持ER图、时序图、流程图等等在项目的releases就可以下载最新版本同时支持在线编辑。_draw github画图

如何开始学习人工智能?入门的学习路径和资源是什么?_人工智能学习路径-程序员宅基地

文章浏览阅读930次。嗨,大家好!如果你对人工智能充满了好奇,并且想要入门这个领域,那么你来对地方了。本文将向你介绍如何从零基础开始学习人工智能,并逐步掌握核心概念和技能。无论你是大学生、职场新人还是对人工智能感兴趣的任何人,都可以按照以下学习路径逐步提升自己。_人工智能学习路径

Unity3D 导入资源_unity怎么导入压缩包-程序员宅基地

文章浏览阅读4.3k次,点赞2次,收藏8次。打开Unity3D的:window-asset store就会出来这样的界面:我们选择一个天空纹理,注意这里的标签只有一个,如果有多个就会显示所有标签的内容:找个比较小的免费的下载一下试试,比如这个:下载以后:点击import就会出现该窗口:然后再点击最底下的import:就导入到我们这里来了。从上面可以切换场景:..._unity怎么导入压缩包

jqgrid 服务器端验证,javascript – jqgrid服务器端错误消息/验证处理-程序员宅基地

文章浏览阅读254次。在你以前的问题的the answer的最后一部分,我试着给出你当前的问题的答案.也许我表示不够清楚.您不应该将错误信息放在标准成功响应中.您应该遵循用于服务器和客户端之间通信的HTTP协议的主要规则.根据HTTP协议实现网格中的加载数据,编辑行和与服务器的所有Ajax通信.每个HTTP响应都有响应第一行的状态代码.了解这个意义非常重要.典型的JSON数据成功请求如下HTTP/1.1 200 OK...._decode message error

白山头讲PV: 用calibre进行layout之间的比对-程序员宅基地

文章浏览阅读4k次,点赞8次,收藏29次。我们在流片之后,通常还是有机会对layout进行局部小的修改。例如metal change eco或者一些层次的局部修改。当我们修改之后,需要进行与之前gds的对比,以便确认没有因为某些..._calibre dbdiff

java exit方法_Java:如何测试调用System.exit()的方法?-程序员宅基地

文章浏览阅读694次。问题我有一些方法应该在某些输入上调用567779278。不幸的是,测试这些情况会导致JUnit终止!将方法调用放在新线程中似乎没有帮助,因为System.exit()终止了JVM,而不仅仅是当前线程。是否有任何常见的处理方式?例如,我可以将存根替换为System.exit()吗?[编辑]有问题的类实际上是一个命令行工具,我试图在JUnit中测试。也许JUnit根本不适合这份工作?建议使用互补回归测..._检查system.exit