【BZOJ4543】Hotel加强版（长链剖分）

最新推荐文章于 2021-03-18 22:35:05 发布

小蒟蒻yyb

最新推荐文章于 2021-03-18 22:35:05 发布

阅读量259

点赞数 1

分类专栏： BZOJ

本文链接：https://blog.csdn.net/qq_30974369/article/details/81582643

版权

BZOJ 专栏收录该内容

382 篇文章 1 订阅

订阅专栏

题面

BZOJ，没有题面
 洛谷，只是普通版本

题解

原来我们的 $O(n^2)$ 做法是设 $f[i][j]$ 表示以 $i$ 为根的子树中，距离 $i$ 的深度为 $j$ 的点的个数，这样子可以每次在 $LCA$ 处合并答案。
然后长链剖分优化一下，就变成了 $O(n)$ 的？？？
玄学的指针我也没太懂啊。。。。~~我才不会说我代码是照着题解打的~~

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define ll long long
#define MAX 100100
inline int read()
{
    int x=0,t=1;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1,n;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int dep[MAX],hson[MAX],md[MAX],a[MAX];
void dfs1(int u,int ff)
{
    md[u]=dep[u]=dep[ff]+1;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff)continue;
        dfs1(v,u);md[u]=max(md[u],md[v]);
        if(md[v]>md[hson[u]])hson[u]=v;
    }
}
ll *f[MAX],*g[MAX],tmp[MAX<<2],*id=tmp,ans;
void dfs(int u,int ff)
{
    if(hson[u])f[hson[u]]=f[u]+1,g[hson[u]]=g[u]-1,dfs(hson[u],u);
    f[u][0]=1;ans+=g[u][0];
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff||v==hson[u])continue;
        f[v]=id;id+=md[v]<<1;g[v]=id;id+=md[v]<<1;
        dfs(v,u);
        for(int j=0;j<md[v];++j)
        {
            if(j)ans+=f[u][j-1]*g[v][j];
            ans+=g[u][j+1]*f[v][j];
        }
        for(int j=0;j<md[v];++j)
        {
            g[u][j+1]+=f[u][j+1]*f[v][j];
            if(j)g[u][j-1]+=g[v][j];
            f[u][j+1]+=f[v][j];
        }
    }
}
int main()
{
    n=read();
    for(int i=1,u,v;i<n;++i)u=read(),v=read(),Add(u,v),Add(v,u);
    dfs1(1,0);f[1]=id;id+=md[1]<<1;g[1]=id;id+=md[1]<<1;
    dfs(1,0);printf("%lld\n",ans);
    return 0;

}