bzoj 3653 谈笑风生——主席树

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3653

原来一直想怎么线段树合并。可是不会把角标挪一位。

查询的其实是子树内一段深度的点的 siz 和。因为是子树内，所以按 dfs 序建立主席树，角标是 dep ，值是 siz 。

注意 long long 。而且数组 *19 就会 RE ，*20就好了。不知为何。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=3e5+,M=N*;
int n,Q,hd[N],xnt,to[N<<],nxt[N<<],siz[N],dfn[N],tim;
int tot,rt[N],ls[M],rs[M],dep[N];
ll sm[M];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void write(ll x)
{
  if(x<)putchar('-'),x=-x;
  if(x<){putchar(x+'');return;}
  write(x/); putchar(x%+'');
}
void add(int x,int y)
{
  to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;
}
void dfs(int cr,int fa)
{
  siz[cr]=; dep[cr]=dep[fa]+;
  for(int i=hd[cr],v;i;i=nxt[i])
    if((v=to[i])!=fa)dfs(v,cr),siz[cr]+=siz[v];
}
void mrg(int l,int r,int &cr,int pr,int p,int k)
{
  cr=++tot; sm[cr]=sm[pr]+k;
  if(l==r)return;  int mid=l+r>>;
  ls[cr]=ls[pr]; rs[cr]=rs[pr];
  if(p<=mid)mrg(l,mid,ls[cr],ls[pr],p,k);
  else mrg(mid+,r,rs[cr],rs[pr],p,k);
}
void dfsx(int cr,int fa)
{
  dfn[cr]=++tim; mrg(,n,rt[dfn[cr]],rt[dfn[cr]-],dep[cr],siz[cr]-);
  for(int i=hd[cr],v;i;i=nxt[i])
    if((v=to[i])!=fa)dfsx(v,cr);
}
ll query(int l,int r,int cr,int pr,int L,int R)
{
  if(l>=L&&r<=R)return sm[cr]-sm[pr];
  int mid=l+r>>; ll ret=;
  if(L<=mid)ret=query(l,mid,ls[cr],ls[pr],L,R);
  if(mid<R)ret+=query(mid+,r,rs[cr],rs[pr],L,R);
  return ret;
}
int main()
{
  n=rdn(); Q=rdn();
  for(int i=,u,v;i<n;i++)
    {
      u=rdn(); v=rdn(); add(u,v); add(v,u);
    }
  dfs(,); dfsx(,);
  for(int i=,p,k;i<=Q;i++)
    {
      p=rdn(); k=rdn();
      ll ans=(ll)min(k,dep[p]-)*(siz[p]-);
      if(siz[p]>)
    ans+=query(,n,rt[dfn[p]+siz[p]-],rt[dfn[p]],dep[p]+,dep[p]+k);
      write(ans); puts("");
    }
  return ;
}