BZOJ4543: [POI2014]Hotel加强版 （长链剖分）

传送门

题解：
记$f_{i,j}$为$i$的子树中深度为$j$的个数，$g_{i,j}$表示$i$的子树中已经匹配的对数， 第三个点到$i$的距离需要$j$。

长链剖分，然后暴力扫非重儿子的子树，这样做是$O(n)$的。 因为每条长链只会被父节点扫一次， 且每条长链不相交。

注意delete释放内存，不然会变慢。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ob==ib)?-1:*ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch==-1) return 0; if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}
const int N=1e5+50;
int n,son[N],mxd[N];
LL ans;
vector <int> edge[N];
deque <LL> *f[N], *g[N];
inline void dfs(int x,int f) {
    for(int e=edge[x].size()-1;e>=0;e--) {
        int v=edge[x][e]; if(v==f) continue;
        dfs(v,x); 
        if(mxd[v]+1>mxd[x]) mxd[x]=mxd[v]+1, son[x]=v;
    }
} 
inline void upt(int x,int p,LL v) {
    ans+=g[x]->operator[](p)*v;
}
inline void add_f(int x,int p,LL v) {
    f[x]->operator[](p)+=v;
}
inline void add_g(int x,int p,LL v) {
    g[x]->operator[](p)+=v;
}
inline void dfs2(int x,int fa) {
    if(son[x]) {
        dfs2(son[x],x);
        f[x]=f[son[x]];
        g[x]=g[son[x]];
        f[x]->push_front(0);
        g[x]->pop_front();
        g[x]->push_back(0);
        g[x]->push_back(0);
    }
    if(f[x]==NULL) f[x]=new deque <LL>, f[x]->push_back(0);
    if(g[x]==NULL) g[x]=new deque <LL>, g[x]->push_back(0);
    upt(x,0,1); add_f(x,0,1);
    for(int e=edge[x].size()-1;e>=0;e--) {
        int v=edge[x][e]; if(v==fa || v==son[x]) continue;
        dfs2(v,x);
        for(int i=0;i<=mxd[v];i++) 
            upt(x,i+1,f[v]->operator[](i));
        for(int i=1;i<=mxd[v];i++)
            upt(v,i,f[x]->operator[](i-1));
        for(int i=0;i<=mxd[v];i++)
            add_g(x,i+1,f[x]->operator[](i+1)*f[v]->operator[](i));
        for(int i=1;i<=mxd[v];i++)
            add_g(x,i-1,g[v]->operator[](i));
        for(int i=0;i<=mxd[v];i++)
            add_f(x,i+1,f[v]->operator[](i));
        delete f[v], delete g[v];
    }
}
int main() {
    n=rd();
    for(int i=1;i<n;i++) {
        int x=rd(), y=rd();
        edge[x].push_back(y);
        edge[y].push_back(x);
    }
    dfs(1,0);
    dfs2(1,0);
    delete f[1], delete g[1];
    cout<<ans;
}