[bzoj3653]谈笑风生 主席树

3653: 谈笑风生

Time Limit: 20 Sec   Memory Limit: 512 MB
[ Submit][ Status][ Discuss]

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

Sample Output

3
1
3

HINT

1<=P<=N

1<=K<=N

N<=300000

Q<=300000

Source

这道题题面都说暴力，但是我们不能暴力

注意a是p号节点，开始没有看到，就是查询 siz[x] * min(dep[p], k)+dfs序在in[p]到out[p]之间且dep在dep[p]+1到dap[p]+k之间的siz和，显然用主席树维护

本来是一次过编译1A的，然而没有加lld

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
const int N = 300000 + 5;
int n,Q,sz,id,cnt;
int last[N],q[N],in[N],out[N],dep[N],siz[N];
int l[N],r[N],root[N],ls[N*20],rs[N*20];
ll sum[N*20],ans;
struct edge{
	int to,next;
}e[N*2];
void insert( int u, int v ){
	e[++cnt].to = v; e[cnt].next = last[u]; last[u] = cnt;
	e[++cnt].to = u; e[cnt].next = last[v]; last[v] = cnt;
}
void dfs( int x, int fa ){
	in[x] = ++id; q[id] = x;
	for( int  i = last[x]; i; i = e[i].next )
		if( e[i].to != fa ){
			dep[e[i].to] = dep[x]+1;
			dfs(e[i].to,x);
			siz[x] += siz[e[i].to]+1;
		}
	out[x] = id;
}
void build( int &k, int l, int r, int x, int val ){
	sum[++sz] = sum[k]+val; ls[sz] = ls[k]; rs[sz] = rs[k]; k = sz;
	if( l == r ) return;
	int mid = (l+r)>>1;
	if( x <= mid ) build( ls[k], l, mid, x, val );
	else build( rs[k], mid+1, r, x, val );
}
ll query( int k, int l, int r, int x, int y ){
	if( y > r ) y = r; if( !k ) return 0;
	if( l == x && r == y ) return sum[k];
	int mid = (l+r)>>1;
	if( y <= mid ) return query( ls[k], l, mid, x, y );
	else if( x > mid ) return query( rs[k], mid+1, r, x, y );
	else return query( ls[k], l, mid, x, mid ) + query( rs[k], mid+1, r, mid+1, y );
}
int main(){
	scanf("%d%d", &n, &Q);
	for( int i = 1,u,v; i < n; i++ ) scanf("%d%d", &u, &v), insert(u,v);
	dfs(1,0); int mx = 0;
	for( int i = 1; i <= n; i++ ) mx = max(mx,siz[i]);
	for( int i = 1; i <= n; i++ ){
		root[i] = root[i-1];
		build( root[i], 0, mx, dep[q[i]], siz[q[i]] );
	}
	while(Q--){
		int p,k; ans = 0;
		scanf("%d%d", &p, &k);
		ans = (ll)siz[p]*min(dep[p],k);
		ans += query(root[out[p]],0,mx,dep[p]+1,dep[p]+k);
		ans -= query(root[in[p]-1],0,mx,dep[p]+1,dep[p]+k);
		printf("%lld\n", ans);
	}
	return 0;
}