首页 > 其他分享 >树分治全家桶

树分治全家桶

时间:2023-12-04 20:59:48浏览次数:30  
标签:int siz 分治 全家 edge maxn mx

树分治全家桶

树,(是种在路边的绿化植物)是图论当中的一种特殊图,(由于绿化环境作用非常优秀)由于特殊性质丰富,经常出现在我们身边。

本文将主要讲述(如何植树)一种树上优美的暴力——树分治。

树分治

树分治可以将部分 \(O(n^2)\) 的暴力降至 \(O(n\log n)\) 级别,尤其适用于树上距离及其的各种变形,树上两点之间路径的问题。

Part1 点分治

处理大规模的树上路径信息问题。

点分治作为树分治的基础思想,主要利用树的重心的不断划分。

例题:P3806 【模板】点分治 1

将以此题为基础,进行讲解。

如果暴力处理每一个询问,复杂度 \(O(n^2)\)。

从树的重心入手,先找出树的重心。

如图的树中,2 节点为树的重心。

那么以 2 为根,遍历整一棵树,求出到各个点的距离,这样就得出了以 2 为一个端点的到各个的点的距离。

我们考虑穿过 2 的一条路径,这样的路径可以由两条不在 2 的同一子树内的路径构成,将这两条路径拼起来即可(相加)。

这样子,所有穿过 2 的(包括以 2 为端点的)路径都求出来了。那么后面求的路径都和 2 无关,直接把 2 删除。

图变为:

然后对于每一新树,我们在分别求出树的重心,重复上述操作,直到树中不存在节点。

对于正确性,可以发现每一条路径都寻找且只寻找了一个路径上的节点进行求路径长度,这样子,每一条路径都经过了统计,正确性可以保证。

对于时间复杂度,根据树的重心的定义,每次新的重心的最大子树大小都至少会比上次找的大小小 \(\frac{1}{2}\),如果我们将次求出的重心向分出这棵子树的重心连边(在上图中,即节点 4、6、5、3 向 2 连边),可以发现,该树的重心的层数为 \(\log n\),而每一层最多只会遍历一个 \(n\) 个节点,时间复杂度达到了优秀的 \(O(n\log n)\)。

例题代码:

#include<bits/stdc++.h>
using namespace std;

const int maxn=2e4+5;

struct node
{
    int to,nxt,w;
}edge[maxn*2];

int n,m,tot,sum,rt,crem;
int dis[maxn],q[maxn],rem[maxn],mx[maxn],siz[maxn],qry[maxn],head[maxn];

bool vis[maxn],jug[maxn*maxn],ans[maxn];
//vis 标记为 1 表示该点删除。
void add(int x,int y,int z)
{
    tot++;
    edge[tot].to=y;
    edge[tot].nxt=head[x];
    edge[tot].w=z;
    head[x]=tot;
}

void dfs_rt(int u,int f)//求重心
{
    mx[u]=0;
    siz[u]=1;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v==f||vis[v]) continue;
        dfs_rt(v,u);
        mx[u]=max(mx[u],siz[v]);
        siz[u]+=siz[v];
    }
    mx[u]=max(sum-siz[u],mx[u]);
    if(mx[rt]>mx[u]||rt==0) rt=u;
}
void dfs_dis(int u,int f)//得距离
{
    rem[++crem]=dis[u];
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]||v==f) continue;
        dis[v]=dis[u]+edge[i].w;
        dfs_dis(v,u);
    }
}

void calc(int u)//求路径信息并合并计算,方法有部分与上述不同
{
    queue<int>que;
    while(!que.empty()) que.pop();
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]) continue;
        crem=0,dis[v]=edge[i].w;
        dfs_dis(v,u);

        for(int j=1;j<=crem;j++)
            for(int k=1;k<=m;k++)
            {
                if(qry[k]>=rem[j]) ans[k]|=jug[qry[k]-rem[j]];
            }

        for(int j=1;j<=crem;j++) que.push(rem[j]),jug[rem[j]]=1;
    }
    while(!que.empty()) jug[que.front()]=0,que.pop();
}

void dfs(int u)//通过重心遍历树
{
    vis[u]=1;
    jug[0]=1;
    calc(u);
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]) continue;
        sum=siz[v],rt=0;
        dfs_rt(v,0);
        dfs(rt);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    for(int i=1;i<=m;i++) scanf("%d",&qry[i]);

    sum=n;
    dfs_rt(1,0);
    dfs(rt);

    for(int i=1;i<=m;i++)
    {
        if(ans[i]) printf("AYE\n");
        else printf("NAY\n");
    }
}
总结

点分治的运用一般和树上两个点有关或者树上的路径有关(一般是存在不定点)。

可以离线处理需要的树上数据。

一般通过离线将一个 \(n\) 消减为 \(\log n\)。

练习

P2634 【国家集训队】 聪聪可可

P3345 【ZJOI2015】 幻想乡战略游戏

Part2 边分治

边分治与点分治类似,不过是冷门算法。

边分治,就是针对边的分治。

选择一条边,这条边分成两边的树的大小最均匀,分别暴力处理两边树的信息,在通过边把所得的信息链接起来,就OK了。

但如果图是菊花图的话边分治会被卡成 \(O(n^2)\)。

咋办哩?

可以把原树转成一棵二叉树,像这样:

Ps:图片出处,OI-wiki树分治章节。

转成二叉树后消除了菊花图的弊端,使得边分治具有普遍性。

由于只会最多增加 \(n\) 个点,总复杂度 \(O(n\log n)\)。

点分治的题边分治也可以做大部分。

Part3 点分树

树分治的究极版本,通过改变原树形态使得层数变为 \(\log n\) 的一种重构树,通常解决与原树形态无关的带修改问题。

它牺牲了原来的结构来换取高速的结构,所以如果子树和父节点有路径(改变距离、一次性更改子树内所有节点、断开边、重连边……)关系,它就完了。

但它支持在线更改部分信息!

还记得点分治的时间复杂度证明吗?在那时,我们将次求出的重心向分出这棵子树的重心连边,证明了不超过 \(\log n\) 层的性质,现在的点分树就是利用这个连边方式,重构整棵树。

原树:

树分治后的重构树:

image-20231204195428226

下面的树没有说明对于的为重构树。

一般来说,我们每一个点会存下到自己的祖先的距离,或者一些其它与祖先相关的信息,点会利用一些数据结构存下关于自己的子树内的信息。

那这种树可以干什么呢?

例题:P6329 【模板】点分树 | 震波

对于本题,可以乱搞。

距离维护一个下标关于距离前缀和(线段树/树状数组),每次改变一个点的操作,可以改变一条该点到根的路径上所有点的前缀和。

对于求值,从该节点 \(v\) 向上遍历祖先 \(x\),求出 \(sum[x][k-dis(v,x)]\),如果是从 \(x\) 的儿子 \(y\) 上来的,要在减去 \(sum[y][k-dis(v,x)-dis(x,y)]\)。

具体参见代码吧。

其实还有很多奇怪的技巧,比如点分治套点分树之类的……

#include<bits/stdc++.h>
using namespace std;

#define inf 1e8

const int maxn=1e5+5;

struct Edge
{
    int to,nxt;
}edge[maxn*2];
struct Tree
{
    int ct;
    int rt[maxn];
    struct node{int ch[2],val;}tree[maxn*55];
    void insert(int &p,int l,int r,int x,int y)
    {
        if(!p) p=++ct;
        if(l==r)
        {
            tree[p].val+=y;
            return ;
        }
        int mid=(l+r)>>1;
        if(x<=mid) insert(tree[p].ch[0],l,mid,x,y);
        else insert(tree[p].ch[1],mid+1,r,x,y);
        tree[p].val=tree[tree[p].ch[0]].val+tree[tree[p].ch[1]].val;
    }
    int query(int p,int l,int r,int ql,int qr)
    {
        if(ql>qr) return 0;
        if(!p) return 0;
        if(ql<=l&&r<=qr) return tree[p].val;
        if(l>qr||r<ql) return 0;
        int mid=(l+r)>>1;
        return query(tree[p].ch[0],l,mid,ql,qr)+query(tree[p].ch[1],mid+1,r,ql,qr);
    }
}w[2];

int n,tot,sum,m,rt;
int val[maxn],head[maxn],mx[maxn],siz[maxn],dis[maxn][25],fa[maxn],K[maxn];

vector<int>E[maxn];

bool vis[maxn];

void Init(int u){sum=siz[u];rt=0;}

void add(int x,int y)
{
    tot++;
    edge[tot].to=y;
    edge[tot].nxt=head[x];
    head[x]=tot;
}

void dfs_rt(int u,int f)
{
    mx[u]=0,siz[u]=1;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]||v==f) continue;
        dfs_rt(v,u);
        siz[u]+=siz[v];
        mx[u]=max(mx[u],siz[v]);
    }
    mx[u]=max(mx[u],sum-siz[u]);
    if(!rt||mx[u]<mx[rt]) rt=u;
}
void dfs_dis(int u,int f,int k)
{
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]||v==f) continue;
        dis[v][k]=dis[u][k]+1;
        dfs_dis(v,u,k);
    }
}
void dfs_calc(int u,int f,int op,int st,int k)
{
    w[op].insert(w[op].rt[st],0,inf,dis[u][k],val[u]);
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]||v==f) continue;
        dfs_calc(v,u,op,st,k);
    }
}

void build(int u)
{
    vis[u]=1;
    K[u]=K[fa[u]]+1;
    dfs_dis(u,0,K[u]);
    dfs_calc(u,0,1,u,K[u]);
    dfs_rt(u,u);
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(vis[v]) continue;
        Init(v);
        dfs_rt(v,u);
        E[u].push_back(rt);
        fa[rt]=u;
        dfs_calc(rt,u,0,rt,K[u]);
        build(rt);
    }
}

void change(int u,int st,int now)
{
    if(!u) return ;
    w[1].insert(w[1].rt[u],0,inf,dis[st][K[u]],now-val[st]);
    if(fa[u]) w[0].insert(w[0].rt[u],0,inf,dis[st][K[u]-1],now-val[st]);
    change(fa[u],st,now);
}
int dfs_ans(int u,int v,int st,int d)
{
    if(!u) return 0;
    if(d-dis[st][K[u]]<0) return dfs_ans(fa[u],u,st,d);
    int res=w[1].query(w[1].rt[u],0,inf,0,d-dis[st][K[u]]);
    if(v) res-=w[0].query(w[0].rt[v],0,inf,0,d-dis[st][K[u]]);
    return res+dfs_ans(fa[u],u,st,d);
}

void outtree(int x)
{
    cout<<x<<endl;
    for(int i=0;i<E[x].size();i++) cout<<x<<" "<<E[x][i]<<endl;
    for(int i=0;i<E[x].size();i++) outtree(E[x][i]);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&val[i]);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }

    siz[1]=n;
    Init(1);
    dfs_rt(1,0);
    build(rt);

    int ans=0;
    for(int i=1;i<=m;i++)
    {
        int op,x,y;
        scanf("%d%d%d",&op,&x,&y);
        x^=ans,y^=ans;
        if(op)
        {
            change(x,x,y);
            val[x]=y;
        }
        else
        {
            printf("%d\n",ans=dfs_ans(x,0,x,y));
        }
    }
}

标签:int,siz,分治,全家,edge,maxn,mx
From: https://www.cnblogs.com/binbinbjl/p/17875931.html

相关文章

  • Vue3快速上手(B站尚硅谷张天禹老师主讲vue全家桶)
    Vue3快速上手Vue3快速上手1.Vue3简介2.Vue3带来了什么1.性能的提升2.源码的升级3.拥抱TypeScript4.新的特性一、创建Vue3.0工程1.使用vue-cli创建2.使用vite创建二、常用CompositionAPI1.拉开序幕的setup2.ref函数3.reactive函数4.Vue3.0中的响应式原理vue2.x的响应式Vue3.0......
  • 重量级消息,微软将ThreadX RTOS全家桶贡献给Eclipse基金会,免费供大家商用,宽松的MIT授权
     从明年第1季度开始,任何人,任何厂家的芯片都可以免费商用,MIT授权就这点好。 贡献出来后,多方可以一起努力开发,当前首批兴趣小组AMD,Cypherbridge,Microsoft,NXP,PX5,Renesas,STMicroelectronics,SiliconLabs,andWitekio(anAvnetcompany)都表达对这个系统支持......
  • 【未完善】多项式全家桶
    #include<iostream>#include<cmath>#include<cctype>#include<functional>#include<algorithm>#include<vector>#defineUP(i,s,e)for(autoi=s;i<e;++i)#defineDOWN(i,e,s)for(autoi=e;i-->s;)usingstd::c......
  • 【刷题记录】20231124 线段树分治
    做题记录:注意到每次相当于\(0\)后面加\(1\),\(1\)后面加\(0\),因此每次可以合并01和10然后将问题规模减半。黑白染色,白格子=lcm+1,黑格子=prime相乘。发现横着竖着有六个质数,斜着只用四个质数。调整一下顺序即可。状压DP。考虑S作为前缀max时S与U-S的排列方案数。S每......
  • 点分治
    点分治是一种在树上进行的分治,可以方便的求解树上路径等问题。例题:P3806【模板】点分治1给定一棵树,询问树上是否存在长度为k的路径。现在我们假设x为根节点,那么一条路径长度为k有两种情况,一种是经过x,一种不经过x,第一种的两个端点在两个不同子树中,第二种的两个端点在同一......
  • 分治与归并
    归并算法:递归+合并,在递归的途中进行分治。递归会把区间越分越小,此时就可以进行分治操作。可以使用全局变量进行分治操作。可以在函数中进行分治操作,在递归树中实现pushup和pushdown,记性父节点与子节点的关系计算。  classSolution{public:structNode{......
  • 苹果电脑 Adobe2023 全家桶 Mac 直装版 最新下载安装
    每一个软件都是亲测上传,都是目前最新的,简化了安装流程适用于小白,全部都是无脑直接安装。Adobe2023全家桶直装版更新日期2023-06-11,包含:AdobeIllustrator、AdobeAcrobatProDC、AdobePremierePro、AdobeAudition、AdobePhotoshop、LightroomClassic、AdobeAfter......
  • 点分治学习笔记(未完成)
    前言点分治不应该算数据结构,它的本质是分治的思想。问题引入对于一个序列\(a\),求是否存在\((l,r)\)使得\(\sum\limits_{i=l}^{r}a_i=k\)。\(n\le10^6,|a_i|\le10^9\)。本题显然是有其它的做法的,由于学的是点分治,所以考虑分治做法。首先对\(a\)求前缀和,记这个数组为......
  • 算法学习笔记(1):CDQ分治
    CDQ分治对比普通分治把一种问题划分成不同子问题,递归处理子问题内部的答案,再考虑合并子问题的答案。再看CDQ分治有广泛的应用,但我不会。但在接下来的题目体会到大概:将可能产生的对于答案的贡献分为两类:\(f(l,mid)\)与\(f(mid+1,r)\)内部产生的贡献,其贡献已在递......
  • 根号分治
    Problem给定一个长度为\(S\)字符串\(s\)与一个正整数\(q\),接下来有\(q\)次询问,第\(i\)次询问给出一个长度为\(T_i\)字符串\(t_i\),求\(t_i\)在\(s\)的出现次数。保证\(S,q,\sum^q_{i=1}T_i\leq10^5\)。Solution后缀自动机,但是我不会。注意到\(\sum^q_{i=1}......