P6778 [Ynoi2009] rpdq gxy001

本文主要参考 IOI 2021 中国国家集训队论文《浅谈一类树分块的构建算法及其应用》周欣。

我们可以通过简单的转化将问题变为求 $(r-l)\sum\limits_{i=l}^r\operatorname{dep}(i)-2\sum\limits_{i=l}^r\sum\limits_{j=i+1}^r\operatorname{dep}(\operatorname{lca}(i,j))$。

显然我们只需要维护后面的部分。

考虑莫队,加入/删除一个元素时产生的贡献是经典题,[LNOI2014]LCA,问题转化为链加链查询和。

我们莫队二次离线,这样我们需要一个支持 $O(\sqrt n)$ 给 $a_i$ 加 $1$,$i$ 取在根到修改点的所有边的编号;$O(1)$ 查询 $\sum a_iw_i$ 的数据结构,$w_i$ 为边权,$i$ 取根到查询点的所有边的编号的数据结构。

我们使用 $\text{top cluster}$ 进行树分块。

定义:

簇是一个联通子图,至多有两个点和全树其他位置连接。

这两个点称为界点,簇内其他点成为内点。

界点间的路径称为簇路径。

树分块可以被认为是把树划分为 $O(\frac nB)$ 个大小为 $O(B)$ 的不交簇。

将界点看为点,簇看为边,形成的树称为收缩树。

构造方法:

$\text{dfs}$ 的时候维护一个栈存放未归类的边。

我们只在三种情况下把当前点作为界点,并将边弹出。

  • 当前点为根节点;
  • 当前点的儿子中存在两个及以上含有界点的子树;
  • 栈中边数量大于 $B$。

问题抽象为给定两个序列 $a$,$b$。将序列划分为若干段使得每个子段 $\sum b$ 不超过 $1$,$\sum a$ 不超过 $B$。

直接贪心地取最长的合法前缀就可以了,正确性证明可以见论文,不在此展开。

回到本题,我们预处理出每个点到它所在簇深度小的界点经过的簇路径的边权和和到它所在簇深度小的界点经过的边权和。

每次修改,把修改点所属簇重新计算每个点到它所在簇深度小的界点经过的边权和,对于其余簇,一定只有簇路径上的边被修改,打标记即可解决,最后在收缩树上做一遍前缀和。

查询时将贡献分为三部分:

  • 收缩树上的前缀和;
  • 簇内的前缀和;
  • 被标记维护的部分,即标记乘“每个点到它所在簇深度小的界点经过的簇路径的边权和”。

直接加起来就可以了。

如果你被卡常,不要惊慌,把 vector 换成手写的内存池或者 basic_string 都能过。

#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#define int unsigned
namespace IO{
	#define BUFSIZE 10000000
	struct read{
		char buf[BUFSIZE],*p1,c;
		read():p1(buf){fread(buf,1,BUFSIZE,stdin);}
		inline read& operator >>(int& x){
			for(c=*p1++,x=0;c!=EOF&&(c<'0'||c>'9');c=*p1++);
			for(;c>='0'&&c<='9';c=*p1++)x=x*10+(c-'0');
			return *this;
		}
	}in;
	struct write{
		char buf[BUFSIZE],*p1,*p2,s[50],f;
		int tp;
		write():p1(buf),p2(buf+BUFSIZE){}
		~write(){fwrite(buf,1,p1-buf,stdout);}
		inline write& operator <<(int x){
			do s[tp++]=x%10+'0',x/=10; while(x);
			while(tp)*p1++=s[--tp];
			return *this;
		}
		inline write& operator <<(char const &x){
			*p1++=x;
			return *this;
		}
	}out;
}
using IO::in;
using IO::out;
int const B=200;
int rt;
std::vector<int> s[200010],er[200010];
int *v[200010],*son[200010],ss[200010],vz[200010],tt[200010],sty[200010],pll[600010],*ed=pll,ctt;
int st[200010],tp,book[200010],dep[200010],bk[200010],bel[200010],bot[200010],a[200010],b[200010],cnt,f[200010],bf[200010],L[200010],R[200010],val[200010];
inline void link(int const &x,int y){
	bf[y]=x,++vz[x];
	tt[ctt]=x,sty[ctt]=y,++ctt;
	while(y!=x)bk[y]=1,y=f[y];
}
void dfs(int const &x,int const &fa){
	int k=tp,ct=0;
	son[x]=ed;
	ed+=s[x].size()-(fa?1:0);
	for(unsigned i=0;i<s[x].size();++i)
		if(s[x][i]!=fa)val[s[x][i]]=er[x][i],son[x][ss[x]++]=s[x][i];
	for(int i=0;i<ss[x];i++){
		int const &to=son[x][i];
		st[tp++]=to,f[to]=x;
		dep[to]=dep[x]+val[to];
		dfs(to,x);
		if(b[to])++ct,b[x]=b[to];
	}
	if(x==rt||ct>1||tp-k>=B){
		book[x]=1;
		b[x]=x;
		int sz=0,lst=ss[x]-1;ct=0;
		for(unsigned i=ss[x]-1;~i;i--){
			sz+=a[son[x][i]];
			if(b[son[x][i]])++ct,link(x,b[son[x][i]]);
			if(sz>B||ct>1){
				sz=a[son[x][i]];
				if(b[son[x][i]])ct=1;
				++cnt;do bel[st[--tp]]=cnt; while(st[tp]!=son[x][i+1]);
				L[cnt]=i+1,R[cnt]=lst,lst=i; 
			}
		}
		++cnt;do bel[st[--tp]]=cnt; while(st[tp]!=son[x][0]);
		L[cnt]=0,R[cnt]=lst; 
	}
	a[x]=tp-k+1;
}
int dis[200010],top[200010],queue[200010],*hd,*tl;
inline void bfs2(){
	hd=tl=queue;
	*tl++=rt;
	while(hd!=tl){
		int const &x=*hd++;
		int d=dis[x];
		if(book[x])d=0;
		for(int i=0;i<ss[x];i++){
			int const &to=son[x][i];
			top[to]=top[x];
			if(bk[to])dis[to]=d+val[to];
			else dis[to]=d;
			top[to]=book[x]?x:top[x];
			*tl++=to;
		}
	}
}
int vv[200010],w[200010],d[200010],tag[200010],p[200010];
inline void update(int x){
	if(x==rt) return;
	int const &k=bel[x];
	do vv[x]+=val[x],x=f[x]; while(!book[x]);
	hd=tl=queue;
	for(int i=L[k];i<=R[k];i++){
		int const &to=son[x][i];
		d[to]=vv[to];
		if(!book[to]) *tl++=to;
	}
	while(hd!=tl){
		int const &x=*hd++;
		for(int i=0;i<ss[x];i++){
			int const &to=son[x][i];
			d[to]=d[x]+vv[to];
			if(!book[to])*tl++=to;
		}
	}
	w[bot[k]]=tag[k]*dis[bot[k]]+d[bot[k]];
	while(x!=rt) ++tag[bel[x]],w[x]+=dis[x],x=bf[x];
	hd=tl=queue;
	*tl++=rt;
	while(hd!=tl){
		int x=*hd++;
		for(int i=0;i<vz[x];i++) p[v[x][i]]=p[x]+w[v[x][i]],*tl++=v[x][i];
	}
}
inline int query(int const &x){return tag[bel[x]]*dis[x]+d[x]+p[top[x]];}
int n,m,p1[200010],p2[200010],BEL[200010];
struct qq{
	int l,r,id;
	inline bool operator <(qq const &x)const{
		return BEL[l]==BEL[x.l]?r<x.r:l<x.l;
	}
}q[200010];
struct off{
	int l,r,k,id;
	off(int const &x,int const &y,int const &d,int const &i):l(x),r(y),k(d),id(i){}
	off():l(),r(),k(),id(){}
};
off *que[200010],pool[400010],*pl=pool;
int ans[200010],sz[200010];
signed main(){
	in>>n>>m;
	rt=1;
	for(int i=1,x,y,z;i<n;i++){
		in>>x>>y>>z;
		s[x].push_back(y),er[x].push_back(z);
		s[y].push_back(x),er[y].push_back(z);
	}
	dfs(rt,0);
	for(int i=1;i<=n;i++)v[i]=ed,ed+=vz[i],vz[i]=0;
	for(int i=0;i<ctt;i++)v[tt[i]][vz[tt[i]]++]=sty[i];
	bfs2();
	for(int i=1;i<=n;i++)if(book[i])bot[bel[i]]=i;
	for(int i=1;i<=n;i++)BEL[i]=(i-1)/B+1;
	for(int i=1;i<=m;i++)in>>q[i].l>>q[i].r,q[i].id=i;
	std::sort(q+1,q+m+1);
	for(int i=1,l=1,r=0;i<=m;i++){
		if(l>q[i].l)++sz[r],l=q[i].l;
		if(r<q[i].r)++sz[l-1],r=q[i].r;
		if(l<q[i].l)++sz[r],l=q[i].l;
		if(r>q[i].r)++sz[l-1],r=q[i].r;
	}
	for(int i=0;i<=n;i++)que[i]=pl,pl+=sz[i],sz[i]=0;
	for(int i=1,l=1,r=0;i<=m;i++){
		if(l>q[i].l)que[r][sz[r]++]=off(q[i].l,l-1,1,q[i].id),l=q[i].l;
		if(r<q[i].r)que[l-1][sz[l-1]++]=off(r+1,q[i].r,-1,q[i].id),r=q[i].r;
		if(l<q[i].l)que[r][sz[r]++]=off(l,q[i].l-1,-1,q[i].id),l=q[i].l;
		if(r>q[i].r)que[l-1][sz[l-1]++]=off(q[i].r+1,r,1,q[i].id),r=q[i].r;
	}
	for(int i=1;i<=n;i++){
		p1[i]=p1[i-1]+query(i);
		update(i);
		for(int k=0;k<sz[i];++k)
			for(int j=que[i][k].l;j<=que[i][k].r;j++)
				ans[que[i][k].id]+=que[i][k].k*query(j);
		p2[i]=p2[i-1]+query(i);
	}
	for(int i=1,l=1,r=0;i<=m;i++){
		if(l>q[i].l)ans[q[i].id]-=p2[l-1]-p2[q[i].l-1],l=q[i].l;
		if(r<q[i].r)ans[q[i].id]+=p1[q[i].r]-p1[r],r=q[i].r;
		if(l<q[i].l)ans[q[i].id]+=p2[q[i].l-1]-p2[l-1],l=q[i].l;
		if(r>q[i].r)ans[q[i].id]-=p1[r]-p1[q[i].r],r=q[i].r;
	}
	for(int i=2;i<=n;i++)dep[i]+=dep[i-1];
	for(int i=2;i<=m;i++)ans[q[i].id]+=ans[q[i-1].id];
	for(int i=1;i<=m;i++)ans[q[i].id]=(q[i].r-q[i].l)*(dep[q[i].r]-dep[q[i].l-1])-2*ans[q[i].id];
	for(int i=1;i<=m;i++)out<<ans[i]<<'\n';
	return 0;
}