正题
题目链接:https://www.lydsy.com/JudgeOnline/problem.php?id=3626
题目大意
一棵树,每次给出(l,r,z)(l,r,z)(l,r,z)询问∑i=lrdepLCA(i,z)\sum_{i=l}^rdep_{LCA(i,z)}i=l∑rdepLCA(i,z)
解题思路
若询问LCA(x,y)LCA(x,y)LCA(x,y)的深度有一种方法,将1∼x1\sim x1∼x的路径都打上标记,然后查询1∼y1\sim y1∼y这条路径上有多少个打了记号的点。
若查询一个点xxx和很多个点的LCALCALCA深度和就可以对于每个点到根节点的路径上+1+1+1,然后查询1∼x1\sim x1∼x路径上数字和即可。
所以我们可以将询问(l,r,z)(l,r,z)(l,r,z)拆分成(1,r,z)−(1,l−1,z)(1,r,z)-(1,l-1,z)(1,r,z)−(1,l−1,z),然后用指针扫过去,用树链剖分维护树上路径和即可。
时间复杂度O((n+Q)log2n)O((n+Q)\log^2n)O((n+Q)log2n)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=5e4+10;
struct Tree_node{ll w,lazy;
};
struct Edge_node{ll to,next;
}a[N];
struct node{ll x,y,val,id;
}q[N*2];
struct Seq_Tree{Tree_node t[N*4];void Downdata(ll x,ll l,ll r){if(!t[x].lazy) return;ll mid=(l+r)/2;t[x*2].w+=t[x].lazy*(mid-l+1);t[x*2+1].w+=t[x].lazy*(r-mid);t[x*2].lazy+=t[x].lazy;t[x*2+1].lazy+=t[x].lazy;t[x].lazy=0;return;}void Change(ll k,ll l,ll r,ll x,ll y,ll val){if(l==x&&r==y){t[k].lazy+=val;t[k].w+=val*(r-l+1);return;}ll mid=(l+r)/2;Downdata(k,l,r);if(y<=mid) Change(k*2,l,mid,x,y,val);else if(x>mid) Change(k*2+1,mid+1,r,x,y,val);else Change(k*2,l,mid,x,mid,val),Change(k*2+1,mid+1,r,mid+1,y,val);t[k].w=t[k*2].w+t[k*2+1].w; }ll Ask(ll k,ll l,ll r,ll x,ll y){if(l==x&&r==y) return t[k].w;ll mid=(l+r)/2;Downdata(k,l,r);if(y<=mid) return Ask(k*2,l,mid,x,y);if(x>mid) return Ask(k*2+1,mid+1,r,x,y);return Ask(k*2,l,mid,x,mid)+Ask(k*2+1,mid+1,r,mid+1,y);}
}T;
ll n,m,cnt,tot,num,ans[N],ls[N],fa[N];
ll siz[N],seq[N],id[N],son[N],dep[N],top[N];
void addl(ll x,ll y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;
}
void dfs1(ll x){siz[x]=1; for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;dep[y]=dep[x]+1;dfs1(y);siz[x]+=siz[y];if(siz[y]>siz[son[x]]) son[x]=y;}return;
}
void dfs2(ll x){id[x]=++cnt;seq[cnt]=x;if(son[x]){top[son[x]]=top[x];dfs2(son[x]);}for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(y==son[x]) continue;top[y]=y;dfs2(y);}
}
void Change_path(ll x,ll y,ll num)
{while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]])swap(x,y); T.Change(1,1,n,id[top[x]],id[x],num);x=fa[top[x]];}if(dep[x]<dep[y]) swap(x,y);T.Change(1,1,n,id[y],id[x],num);
}
ll Ask_path(ll x,ll y)
{ll ans=0;while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]])swap(x,y); ans+=T.Ask(1,1,n,id[top[x]],id[x]);x=fa[top[x]];}if(dep[x]<dep[y]) swap(x,y);ans+=T.Ask(1,1,n,id[y],id[x]); return ans;
}
bool cmp(node x,node y)
{return x.x<y.x;}
int main()
{scanf("%lld%lld",&n,&m);for(ll i=2;i<=n;i++){scanf("%lld",&fa[i]);fa[i]++;addl(fa[i],i);}dep[1]=1;dfs1(1);dfs2(1);for(ll i=1;i<=m;i++){ll l,r,z;scanf("%lld%lld%lld",&l,&r,&z);l++;r++;z++;q[++num]=(node){l-1,z,-1,i};q[++num]=(node){r,z,1,i};}sort(q+1,q+1+num,cmp);ll now=1;for(ll i=1;i<=num;i++){while(now<=q[i].x)Change_path(1,now,1),now++;ans[q[i].id]+=q[i].val*Ask_path(1,q[i].y);}for(ll i=1;i<=m;i++)printf("%lld\n",ans[i]%201314);
}