题目在这里呀!
个人认为是一道很好的题目,原来可持久化线段树还能这么用,看题解之前还是没有想到啦要批评!那就写个题解补偿一下?
题意
给你一棵有根树,n个节点,有q次询问,每次询问,给出两个数x(1<=x<=n),d,求有多少有序元组(y,z)满足 x,y,z互不相同,x,y均为z的祖先,且x,y之间的距离超过d。
题解
y的位置有两种情况
1、y是x的祖先
那么y一定是在x到根节点的路径上,有min(dep[x],d)种取法,然后z有size[x]种取法,故方案数为size[x]*min(dep[x],d)。
2、y是x的孩子
那么方案数为
∑isize[i]
∑
i
s
i
z
e
[
i
]
(dep[i]-dep[x]<=d)
怎么求这个?考虑按dfs序来建可持久化线段树,然后深度来作为节点编号,节点里放size[]。吼吼吼为什么一开始没有想到没有想到?!
然后就迎刃而解啦(还傻傻地忘记写了一个return被嘲笑qwq)
//Suplex
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#define ll long long
#define N 400400
using namespace std;
int n,Q,u,v,cnt,tot,vet[N+N],Next[N+N],head[N],L[N],R[N],opp[N],dep[N],size[N],p,k,maxd;
int root[N];
struct Segment_tree{
int l,r;
ll val;
}t[8000000];
void add_edge(int u,int v)
{
vet[++cnt]=v;Next[cnt]=head[u];head[u]=cnt;
}
void dfs(int u,int fa)
{
L[u]=++tot;opp[tot]=u;
for(int i=head[u];i;i=Next[i]){
int v=vet[i];
if(v==fa) continue;
dep[v]=dep[u]+1;
dfs(v,u);
size[u]+=size[v]+1;
}
R[u]=tot;
}
inline void modify(int &rt,int p,int l,int r,int x,int delta)
{
rt=++cnt;
t[rt].l=t[p].l;t[rt].r=t[p].r;
t[rt].val=t[p].val+delta;
if(l==r) return;
int mid=(l+r)>>1;
if(x<=mid) modify(t[rt].l,t[p].l,l,mid,x,delta);
else modify(t[rt].r,t[p].r,mid+1,r,x,delta);
}
inline ll query(int rt,int l,int r,int x,int y)
{
if(!rt) return 0;
if(x<=l && r<=y) return t[rt].val;
int mid=(l+r)>>1;
if(y<=mid) return query(t[rt].l,l,mid,x,y);
else if(x>mid) return query(t[rt].r,mid+1,r,x,y);
else return query(t[rt].l,l,mid,x,mid)+query(t[rt].r,mid+1,r,mid+1,y);
}
int main()
{
scanf("%d%d",&n,&Q);
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
add_edge(u,v);add_edge(v,u);
}
dfs(1,-1);
for(int i=1;i<=n;i++) maxd=max(maxd,dep[i]);
cnt=0;
for(int i=1;i<=n;i++) modify(root[i],root[i-1],0,maxd,dep[opp[i]],size[opp[i]]);
while(Q--){
scanf("%d%d",&p,&k);
ll ans=1ll*size[p]*min(dep[p],k);
ans+=query(root[R[p]],0,maxd,dep[p]+1,dep[p]+k);
ans-=query(root[L[p]-1],0,maxd,dep[p]+1,dep[p]+k);
printf("%lld\n",ans);
}
return 0;
}