Description
设T 为一棵有根树,我们做如下的定义:
• 设a和b为T 中的两个不同节点。如果a是b的祖先,那么称“a比b不知道
高明到哪里去了”。
• 设a 和 b 为 T 中的两个不同节点。如果 a 与 b 在树上的距离不超过某个给定
常数x,那么称“a 与b 谈笑风生”。
给定一棵n个节点的有根树T,节点的编号为1 到 n,根节点为1号节点。你需
要回答q 个询问,询问给定两个整数p和k,问有多少个有序三元组(a;b;c)满足:
1. a、b和 c为 T 中三个不同的点,且 a为p 号节点;
2. a和b 都比 c不知道高明到哪里去了;
3. a和b 谈笑风生。这里谈笑风生中的常数为给定的 k。
Input
输入文件的第一行含有两个正整数n和q,分别代表有根树的点数与询问的个数。接下来n - 1行,每行描述一条树上的边。每行含有两个整数u和v,代表在节点u和v之间有一条边。
接下来q行,每行描述一个操作。第i行含有两个整数,分别表示第i个询问的p和k。
Output
输出 q 行,每行对应一个询问,代表询问的答案。
Sample Input
5 3
1 2
1 3
2 4
4 5
2 2
4 1
2 3
1 2
1 3
2 4
4 5
2 2
4 1
2 3
Sample Output
3
1
3
HINT
1<=P<=N
1<=K<=N
N<=300000
Q<=300000
题解:
首先询问1很好处理.
对于询问2,对dfs序建主席树即可.每次询问就是查询p所在子树中,深度在[deep[p]+1,deep[p]+k]的节点的size和.
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#define LL long long
#define N 300010
using namespace std;
int n,Q,k,p,x,y,point[N],next[N<<1],sz,cnt,num,mx,in[N],out[N],deep[N],q[N];
int root[N],ls[N*20],rs[N*20];
LL sum[N*20],ans,size[N];
struct use{int st,en;}e[N<<1];
void add(int x,int y){next[++cnt]=point[x];point[x]=cnt;e[cnt].en=y;}
int read(){
int x=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
void dfs(int x,int fa){
in[x]=++num;q[num]=x;
for(int i=point[x];i;i=next[i])
if(e[i].en!=fa){
deep[e[i].en]=deep[x]+1;mx=max(mx,deep[e[i].en]);
dfs(e[i].en,x);size[x]+=size[e[i].en]+1;
}
out[x]=num;
}
void insert(int x,int &y,int l,int r,int p,LL v){
y=++sz;sum[y]=sum[x]+v;
if(l==r)return;
ls[y]=ls[x];rs[y]=rs[x];
int mid=(l+r)>>1;
if(p<=mid)insert(ls[x],ls[y],l,mid,p,v);
else insert(rs[x],rs[y],mid+1,r,p,v);
}
LL query(int x,int y,int l,int r,int ll,int rr){
rr=min(rr,r);if (ll>rr) return 0;
if(l==ll&&r==rr) return sum[y]-sum[x];
int mid=(l+r)>>1;
if(rr<=mid)return query(ls[x],ls[y],l,mid,ll,rr);
else if(mid<ll)return query(rs[x],rs[y],mid+1,r,ll,rr);
else return query(ls[x],ls[y],l,mid,ll,mid)+query(rs[x],rs[y],mid+1,r,mid+1,rr);
}
int main(){
n=read();Q=read();
for(int i=1;i<n;i++){
x=read();y=read();
add(x,y);add(y,x);
}
dfs(1,0);
for(int i=1;i<=n;i++) insert(root[i-1],root[i],0,mx,deep[q[i]],size[q[i]]);
for(int i=1;i<=Q;i++) {
p=read();k=read();ans=0;
ans+=size[p]*(LL)min(deep[p],k);
ans+=query(root[in[p]-1],root[out[p]],0,mx,deep[p]+1,deep[p]+k);
printf("%lld\n",ans);
}
}