题意
给出一个n个节点的有根树(编号为0到n-1,根节点为0)。一个点的深度定义为这个节点到根的距离+1。
设dep[i]表示点i的深度,LCA(i,j)表示i与j的最近公共祖先。
有q次询问,每次询问给出l r z,求sigma_{l<=i<=r}dep[LCA(i,z)]。
(即,求在[l,r]区间内的每个节点i与z的最近公共祖先的深度之和)
解析
首先发现一个性质,加上depth[LCA(l,r)]可以等价于将根到l加上1,之后查询根到r。扩展到区间的话,设要求结点为z,区间[l,r]。也就是从1加到r后对z的查询值减去1加到l-1对z的查询值。这样的话可以把询问先离线排序,之后树链剖分维护,在标记一下就可以了。
#include <cstdio>
#include <algorithm>
#define Rep( i , B , E ) for(int i=B;i<=E;i++)
#define For( i , B , E ) for(int i=B;i;i=E)
using namespace std;
const int maxx = 100000 + 25;
const int mod = 201314;
int head[maxx],to[maxx<<1],nxt[maxx<<1];
int top[maxx],ftr[maxx],son[maxx],rnk[maxx],size[maxx],dpt[maxx];
int T[maxx<<2],Add[maxx<<2],Ans[maxx];
int n,m,x,y,z,num,cnt,tot,pos = 1;
namespace Y{
void Ins(int x,int y) {to[++num]=y;nxt[num]=head[x];head[x]=num;}