BZOJ3653: 谈笑风生

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

题解：稍微思考可以知道只要求x子树内dep[y]-dep[x]<=k 的 s[y]-1 之和即可。

我刚开始想得是 线段树/树状数组 套 treap，显然可做，但是nlog^2n ，显然要T。。。

于是我们转化思路，直接用一维dep来建主席树，这样子树内的信息都是O（1）就可以知道了。

查询区间和只要和普通线段树一样即可O（logn）。

真是一道好题！

代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 300000+5

#define maxm 6000000+5

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go)

#define for5(n,m) for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)

#define mod 1000000007

using namespace std;

inline int read()

{

    int x=0,f=1;char ch=getchar();

    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}

    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,m,head[maxn],tot,cnt,ti,id[maxn],pos[maxn][2],rt[maxn],s[maxn],dep[maxn];
struct edge{int go,next;}e[2*maxn];
int ls[maxm],rs[maxm];
ll sum[maxm];
inline void add(int x,int y)
{
    e[++tot]=(edge){y,head[x]};head[x]=tot;
    e[++tot]=(edge){x,head[y]};head[y]=tot;
}
inline void dfs(int x)
{
    id[pos[x][0]=++ti]=x;s[x]=1;
    for4(i,x)if(!pos[y][0])dep[y]=dep[x]+1,dfs(y),s[x]+=s[y];
    pos[x][1]=ti;
}
inline void update(int l,int r,int x,int &y,int z,int w)
{
    y=++cnt;
    sum[y]=sum[x]+(ll)w;
    if(l==r)return;
    int mid=(l+r)>>1;
    ls[y]=ls[x];rs[y]=rs[x];
    if(z<=mid)update(l,mid,ls[x],ls[y],z,w);else update(mid+1,r,rs[x],rs[y],z,w);
}
inline ll query(int x,int y,int l,int r,int xx,int yy)
{
    if(l==xx&&r==yy)return sum[y]-sum[x];
    int mid=(l+r)>>1;
    if(yy<=mid)return query(ls[x],ls[y],l,mid,xx,yy);
    else if(xx>mid)return query(rs[x],rs[y],mid+1,r,xx,yy);
    else return query(ls[x],ls[y],l,mid,xx,mid)+query(rs[x],rs[y],mid+1,r,mid+1,yy);
}

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();m=read();
    for1(i,n-1)add(read(),read());
    dfs(1);
    for1(i,n)update(0,n,rt[i-1],rt[i],dep[id[i]],s[id[i]]-1);
    while(m--)
    {
        int x=read(),y=read();
        printf("%lld\n",(ll)(s[x]-1)*min(dep[x],y)+query(rt[pos[x][0]-1],rt[pos[x][1]],0,n,dep[x]+1,min(dep[x]+y,n)));
    }

    return 0;

}  

 一开始一直WA是因为没有把0建到主席树里。