[湖南集训]谈笑风生 长链剖分

[湖南集训]谈笑风生

LG传送门

这题有很多方法做，一堆大佬用线段树合并、主席树来做，但是我太弱了，只会长链剖分。

长链剖分的解法，思维简单，码量极低，是你的不二之选！

还是老规矩，先上\(n^2\)DP。

题意就不说了，但是先注意到，如果\(b\)在\(a\)的上方，\(c\)直接取\(a\)子树中除了\(a\)的所有点就好了，DP的目的只是统计\(b\)在\(a\)的下方的情况。

首先设\(f[i][j]\)表示如果把\(a\)取在\(i\)点，\(b\)取在以\(i\)为根的子树中距离\(a\)为\(j\)的点上的方案数。特别地，\(f[i][0]\)表示把\(b\)取在\(a\)点上的方案数（其实就是子树内除了\(a\)的所有点），然而实际上题目要求\(a\)和\(b\)不是同一个点，记答案的时候需要减去它的贡献，这样定义只是为了方便转移。

可以写出转移方程：\(f[i][j+1]=\sum f[t][j]\)，其中\(t\)是\(i\)的子节点。

观察一下题目要求的东西，发现需要对\(f[i][j]\)求个前缀和（意思是对于同一个\(i\)，求\(j\)变化时的前缀和），然而前缀和不方便，那就求后缀和。

所以我们的\(f[i][j]\)的意义需要在原来的基础上做一些调整，变成原来意义上的后缀和。方程没有变：\(f[i][j+1]=\sum f[t][i]\)，只是加上一句\(f[i][0]+=f[t][0]\)就好了。

DP完一个点之后就统计答案，很简单就不赘述了。

但是上面这个DP是\(O(n^2)\)的，由于下标是深度，套个长链剖分的板子就好了。不会长链剖分的同学可以看看我的长链剖分总结。

#include<cstdio>
#include<cctype>
#include<algorithm>
#include<vector>
#define R register
#define I inline
using namespace std;
const int S=300003,N=600003;
char buf[S],*p1,*p2;
I char gc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,S,stdin),p1==p2)?EOF:*p1++;}
I int rd(){
    R int f=0; R char c=gc();
    while(c<48||c>57) c=gc();
    while(c>47&&c<58) f=f*10+(c^48),c=gc();
    return f;
}
int h[S],s[N],g[N],d[S],w[S],t[S],p[S],q[S],c; long long *f[S],u[S],o[S],*e=u+1;
vector<pair<int,int> > v[S];
I int min(int x,int y){return x<y?x:y;}
I void add(int x,int y){s[++c]=h[x],g[c]=y,h[x]=c;}
I void ini(int x){f[x]=e,e+=t[x]-d[x]+1;}
void dfs1(int x,int f){t[x]=d[x]=d[f]+1,p[x]=f,w[x]=1;
    for(R int i=h[x],y;i;i=s[i])
        if((y=g[i])^f){dfs1(y,x),w[x]+=w[y];
            if(t[y]>t[x]) t[x]=t[y],q[x]=y;
        }
}
void dfs2(int x){f[x][0]=w[x]-1;
    if(q[x]) f[q[x]]=f[x]+1,dfs2(q[x]),f[x][0]+=f[q[x]][0];
    R int i,j,k,y,m;
    for(i=h[x];i;i=s[i])
        if((y=g[i])^p[x]&&y^q[x])
            for(j=0,ini(y),dfs2(y),f[x][0]+=f[y][0],m=t[y]-d[y];j<=m;++j)
                f[x][j+1]+=f[y][j];
    for(i=0,m=v[x].size()-1;i<=m;++i){
        k=v[x][i].first,j=v[x][i].second,o[j]+=1ll*(w[x]-1)*min(d[x]-1,k);
        if(k>=t[x]-d[x]) o[j]+=f[x][0]-w[x]+1;
        else o[j]+=f[x][0]-w[x]+1-f[x][k+1];
    }
}
int main(){
    R int n=rd(),m=rd(),i,x,y;
    for(i=1;i<n;++i) x=rd(),y=rd(),add(x,y),add(y,x);
    for(dfs1(1,0),i=1;i<=m;++i) x=rd(),y=rd(),v[x].push_back(make_pair(y,i));
    for(ini(1),dfs2(1),i=1;i<=m;++i) printf("%lld\n",o[i]);
    return 0;
}

转载于:https://www.cnblogs.com/cj-chd/p/10076228.html