【BZOJ 3653】 谈笑风生|主席树

好神的思路

按dfs序建树 深度作为范围

详见黄学长等

wa了一次

1.没有 long long 输出

2.询问的时候没有加root[]调试很长时间

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
const int MAXN=300000+10;
int cnt,s[MAXN],t[MAXN],size[MAXN],depth[MAXN],fa[MAXN],fx[MAXN];
int top,root[MAXN],max_depth,to[MAXN*20][2];
LL Num[MAXN*20];
int tot,g[MAXN],nnext[MAXN*2],num[MAXN*2];
void add(int x,int y)
{
tot++;
nnext[tot]=g[x];
g[x]=tot;
num[tot]=y;
}
void dfs(int x)
{
depth[x]=depth[fa[x]]+1;
max_depth=max(max_depth,depth[x]);
s[x]=++cnt;
fx[cnt]=x;
size[x]=1;
for(int i=g[x];i;i=nnext[i])
if(num[i]!=fa[x])
{
fa[num[i]]=x;
dfs(num[i]);
size[x]+=size[num[i]];
}
t[x]=cnt;
}
int n,q;
void add(int last,int now,int l,int r,int x,int y)
{
Num[now]=Num[last]+y;
if(l==r) return ;
int mid=(l+r)/2;
if(x<=mid) to[now][1]=to[last][1],to[now][0]=++top,add(to[last][0],to[now][0],l,mid,x,y);
else to[now][0]=to[last][0],to[now][1]=++top,add(to[last][1],to[now][1],mid+1,r,x,y);
}
LL qq(int last,int now,int l,int r,int ss,int tt)
{
// cout<<l<<" "<<r<<endl;
if(ss<=l&&r<=tt) return Num[now]-Num[last];
int mid=(l+r)/2;
LL ans=0;
if(ss<=mid) ans+=qq(to[last][0],to[now][0],l,mid,ss,tt);
if(mid+1<=tt) ans+=qq(to[last][1],to[now][1],mid+1,r,ss,tt);
return ans;
}
int main()
{
cin>>n>>q;
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d %d",&x,&y);
add(x,y);
add(y,x);
}
dfs(1);
for(int i=1;i<=n;i++) root[i]=++top,add(root[i-1],root[i],1,max_depth,depth[fx[i]],size[fx[i]]-1);
// for(int i=1;i<=n;i++) cout<<Num[root[i]]<<' ';
//cout<<endl;
//cout<<'*'<<qq(root[3],root[5],1,max_depth,1,4);

for(int i=1;i<=q;i++)
{
LL ans=0;
int x,y;
scanf("%d %d",&x,&y);
ans+=((LL)min(y,depth[x]-1)*(size[x]-1));//cout<<s[x]<<' '<<t[x]<<endl;
ans+=qq(root[s[x]],root[t[x]],1,max_depth,depth[x]+1,depth[x]+y);
printf("%lld\n",ans);
}
return 0;
}