BZOJ3730: 震波

BZOJ3730: 震波

https://lydsy.com/JudgeOnline/problem.php?id=3730

分析:

• 对于点分树上的每个点\(x\)，维护子树所有点到\(x\)的距离和到点分树上\(fa[x]\)的距离 。
• 查询时查询\(x\)在点分树上到根的一条链，容斥计算答案。
• 然后因为有修改我们使用树状数组。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <iostream>
using namespace std;
#define N 100050
char buf[100000],*p1,*p2;
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {
	int x=0;
	char s=nc();
	while(s<'0') s=nc();
	while(s>='0') x=(((x<<2)+x)<<1)+s-'0',s=nc();
	return x;
}
#define db(x) cerr<<#x<<" = "<<x<<endl
int head[N],to[N<<1],nxt[N<<1],cnt,n,m,val[N];
int siz[N],fk[N],root,tot,used[N],fa[N][20],dis[N][20],dep[N];
struct Bit {
	vector<int>b;
	void make(int s) {b.resize(s+2);}
	void fix(int x,int v) {
		x++;
		int lim=b.size();
		for(;x<lim;x+=x&(-x)) b[x]+=v;
	}
	int inq(int x) {
		x++;
		int re=0;
		x=min(x,int(b.size())-1);
		for(;x;x-=x&(-x)) re+=b[x];
		return re;
	}
}f1[N],f2[N];
inline void add(int u,int v) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void gr(int x,int y) {
	int i;
	siz[x]=1; fk[x]=0;
	for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {
		gr(to[i],x); siz[x]+=siz[to[i]];
		fk[x]=max(fk[x],siz[to[i]]);
	}
	fk[x]=max(fk[x],tot-siz[x]);
	if(fk[root]>fk[x]) root=x;
}
int mxd;
void gd(int x,int y,int rt,int d) {
	int i;
	mxd=max(mxd,d);
	for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {
		mxd=max(mxd,dis[to[i]][dep[to[i]]]);
		fa[to[i]][++dep[to[i]]]=rt;
		dis[to[i]][dep[to[i]]]=d;
		gd(to[i],x,rt,d+1);
	}
}
void solve(int x) {
	used[x]=1;
	int i;
	mxd=0;
	gd(x,0,x,1);
	f1[x].make(mxd); f2[x].make(mxd);
	int all=tot;
	for(i=head[x];i;i=nxt[i]) if(!used[to[i]]) {
		tot=siz[to[i]]; 
		root=0; gr(to[i],x); solve(root);
	}
}
void upd(int x,int v) {
	int i;
	f1[x].fix(0,v);
	f2[x].fix(dis[x][dep[x]],v);
	for(i=dep[x];i;i--) {
		f1[fa[x][i]].fix(dis[x][i],v);
		f2[fa[x][i]].fix(dis[x][i-1],v);
	}
}
int query(int x,int k) {
	int i,re=f1[x].inq(k);
	for(i=dep[x];i;i--) {
		if(k>=dis[x][i]) {
			re+=f1[fa[x][i]].inq(k-dis[x][i]);
			re-=f2[fa[x][i+1]].inq(k-dis[x][i]);
		}
	}
	return re;
}
int main() {
	n=rd(),m=rd();
	int i,x,y,opt;
	for(i=1;i<=n;i++) val[i]=rd();
	for(i=1;i<n;i++) {
		x=rd(), y=rd();
		add(x,y); add(y,x);
	}
	tot=n;
	fk[0]=1ll<<30;
	gr(1,0);
	solve(root);
	for(i=1;i<=n;i++) fa[i][dep[i]+1]=i;
	for(i=1;i<=n;i++) upd(i,val[i]);
	int ans=0;
	while(m--) {
		opt=rd(), x=rd(), y=rd();
		x^=ans, y^=ans;
		if(!opt) {
			ans=query(x,y);
			printf("%d\n",ans);
		}else {
			upd(x,y-val[x]), val[x]=y;
		}
	}
}