【BZOJ3653】談笑風生（長鏈剖分）

【BZOJ3653】談笑風生（長鏈剖分）

題面

BZOJ
洛谷
權限題啊。。。。

題解

首先根據題目給的條件，發現\(a,b\)都要是\(c\)的父親。
因此這三個點是樹上的一條深度單增的鏈。
由於\(a,b\)之間距離不超過\(k\)，而且\(a\)被欽定了，因此只有兩種狀況：
一種是\(a\)是\(b\)的祖先，貢獻是\(\sum_b size[b]-1\)，也就是全部\(b\)能夠選擇的點的子樹和。
另一種\(b\)是\(a\)的祖先，貢獻是\(\sum_b size[a]-1\)，欽定一個\(b\)以後，\(c\)能夠在\(a\)的子樹中任選。
第二種狀況很簡單，由於\(size[a]\)是定值，而且每一個點的父親是惟一的，因此第二部分很容易算。
困難的是第一部分，然而依舊不難把。。。
方法不少，好比說，你把\(dfs\)序和深度當作\(x,y\)軸，這樣子就是二維數點，直接主席樹。
或者說直接點分治也能夠。

固然，既然想寫長鏈剖分，那就固然要用長鏈剖分來作啊。
咱們發現，全部的值都由重兒子向後挪動一位得來，而咱們要求的東西須要維護一個區間和。
這樣子很很差用前綴和來作，因此咱們能夠用一個後綴和啊！
這樣子就很舒服了，直接維護後綴和，而後長鏈剖分轉移，能夠作到複雜度\(O(n)\)。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define ll long long
#define MAX 300300
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
struct Ask{int id,k;};
vector<Ask> p[MAX];
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
ll tmp[MAX],*s[MAX],*id=tmp,ans[MAX];
int md[MAX],dep[MAX],fa[MAX],hson[MAX],size[MAX];
void dfs1(int u,int ff)
{
    md[u]=dep[u]=dep[ff]+1;fa[u]=ff;size[u]=1;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff)continue;
        dfs1(v,u);size[u]+=size[v];
        if(md[v]>md[hson[u]])hson[u]=v;
    }
    if(hson[u])md[u]=md[hson[u]];
}
void dfs(int u)
{
    s[u][0]=size[u]-1;
    if(hson[u])s[hson[u]]=s[u]+1,dfs(hson[u]),s[u][0]+=s[hson[u]][0];
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==fa[u]||v==hson[u])continue;
        s[v]=id;id+=md[v]-dep[v]+1;dfs(v);
        for(int j=0;j<=md[v]-dep[v];++j)s[u][j+1]+=s[v][j];
        s[u][0]+=s[v][0];
    }
    for(int i=p[u].size()-1;~i;--i)
    {
        int k=p[u][i].k,id=p[u][i].id;
        ans[id]+=1ll*(size[u]-1)*min(dep[u]-1,k);
        if(k>=md[u]-dep[u])ans[id]+=s[u][0]-size[u]+1;
        else ans[id]+=s[u][0]-size[u]+1-s[u][k+1];
    }
}
int n,Q;
int main()
{
    n=read();Q=read();
    for(int i=1;i<n;++i)
    {
        int u=read(),v=read();
        Add(u,v);Add(v,u);
    }
    dfs1(1,0);
    for(int i=1;i<=Q;++i)
    {
        int u=read(),k=read();
        p[u].push_back((Ask){i,k});
    }
    s[1]=id;id+=md[1];dfs(1);
    for(int i=1;i<=Q;++i)printf("%lld\n",ans[i]);
    return 0;
}