长链剖分学习笔记

最新推荐文章于 2020-03-30 16:58:55 发布

weixin_30467087

最新推荐文章于 2020-03-30 16:58:55 发布

阅读量107

点赞数

原文链接：http://www.cnblogs.com/wo-shi-zhen-de-cai/p/11580906.html

版权

~~终于认真写一次标题了~~

因为一些不明原因，之前对\(dsu\) \(on\) \(tree\)的理解没有完全写出来，在这里会一起写，因为两者极为相似。

先来看一下\(dsu\) \(on\) \(tree\)和长链剖分的对比。

\(dsu\) \(on\) \(tree\)实际上就是重链剖分，可以处理很多与子树有关且不带修改的题目（离线），复杂度：\(\mathcal{O}(nlogn)\)。

长链剖分实际上就是长链剖分，可以处理很多与深度有关且不带修改的题目（离线），复杂度：\(\mathcal{O}(n)\)。

可以看出，长链剖分比\(dsu\) \(on\) \(tree\)适用范围更小，但更优秀。（具体原因后面会口胡）

考虑一下两者的暴力，即为开一个数组维护信息，每次跑一边子树维护信息，再统计答案，复杂度显然\(\mathcal{O}(n^2)\)的。

实际上，我们的暴力相当于每次跑子树时清空一遍数组，再维护信息。（原暴力显然可以在线，后面会不动声色变为离线，方便我们优化）

因此，我们考虑保留，再将其他信息加入，可以优化。

如果信息和深度无关，考虑重链剖分，先跑轻儿子，并处理他的答案，注意这里我们要清除信息，再跑重儿子，保留信息，处理答案，再暴力扫一遍轻儿子维护信息即可。

而当信息和深度有关时，显然用长链剖分更优秀些，通过在一个信息数组上的覆盖，完成了上面的清除信息（此处指针实现。

例题：\(Cnoi2019\)雪松果树

板子，直接上代码，重点是这题卡空间，倍增过不了，\(Vector\)也有点悬……

#include<bits/stdc++.h>
using namespace std;
inline int read()
{
    int f=1,w=0;char x=0;
    while(x<'0'||x>'9') {if(x=='-') f=-1; x=getchar();}
    while(x!=EOF&&x>='0'&&x<='9') {w=(w<<3)+(w<<1)+(x^48);x=getchar();}
    return w*f;
}
const int N=1000010;
int num_edge,n,q,Top,ans[N];
int head[N],Dep[N],Max[N],Son[N];
int Stk[N],Tmp[N<<2],*f[N],*Now=Tmp;
vector<pair<int,int> > Que[N],Anc[N];
struct Edge{int next,to;} edge[N];
inline void Add(int from,int to)
{
    edge[++num_edge].next=head[from];
    edge[num_edge].to=to;
    head[from]=num_edge;
}
inline void Dfs_For_Pre(int pos)
{
    for(int i=head[pos];i;i=edge[i].next)
    {
        Dfs_For_Pre(edge[i].to);
        if(Max[Son[pos]]<Max[edge[i].to]) Son[pos]=edge[i].to;
    }
    Max[pos]=Max[Son[pos]]+1;
}
inline void Dfs_For_Anc(int pos)
{
    Stk[++Top]=pos;
    for(int i=0;i<(int)Que[pos].size();i++)
        if(Top>Que[pos][i].first)
            Anc[Stk[Top-Que[pos][i].first]].push_back(Que[pos][i]);
    for(int i=head[pos];i;i=edge[i].next) Dfs_For_Anc(edge[i].to);
    Top--;
}
inline void Long_Chain_Div(int pos)
{
    for(int i=head[pos];i;i=edge[i].next)
        if(edge[i].to!=Son[pos])
            f[edge[i].to]=Now,Now+=Max[edge[i].to],Long_Chain_Div(edge[i].to);
    if(Son[pos]) f[Son[pos]]=f[pos]+1,Long_Chain_Div(Son[pos]);f[pos][0]=1;
    for(int i=head[pos];i;i=edge[i].next)
        if(edge[i].to!=Son[pos])
            for(int j=1;j<=Max[edge[i].to];j++)
                f[pos][j]+=f[edge[i].to][j-1];
    for(int i=0;i<(int)Anc[pos].size();i++)
        ans[Anc[pos][i].second]=f[pos][Anc[pos][i].first]-1;
}
int main(){
#ifndef ONLINE_JUDGE
    freopen("A.in","r",stdin);
#endif
    n=read(),q=read();
    for(int i=2,x;i<=n;i++) x=read(),Add(x,i);
    for(int i=1,x,k;i<=q;i++)
        x=read(),k=read(),Que[x].push_back(make_pair(k,i));
    Dfs_For_Anc(1);Dfs_For_Pre(1);
    f[1]=Now;Now+=Max[1]+1;Long_Chain_Div(1);
    for(int i=1;i<=q;i++) printf("%d ",ans[i]);
}

转载于:https://www.cnblogs.com/wo-shi-zhen-de-cai/p/11580906.html