树上算法

树上算法

最近公共祖先(LCA)

  • \(\operatorname{lca}(u,v)\)为点 \(u\)\(v\) 的所有公共祖先中最深的一个

  • 暴力跳父亲 \(O(n)\)

    倍增 \(O(n\log n)\)预处理 \(O(\log n)\) 查询

    树链剖分 \(O(n)\) 预处理 \(O(\log n)\)

    \(\operatorname{Tarjan}\) 均摊\(O(1)\)(离线)

    \(\operatorname{Rmq}\) \(O(n \log n)\) 预处理,\(O(1)\)查询

  • 倍增法

    • \(f(i,j)\) 代表 \(i\)\(2^j\) 级父亲

      易知 \(f(i,j)=f(f(i,j-1),j-1)\)

      即“ \(i\)\(2^j\) 级父亲 为 \(i\)\(2^{j-1}\) 级父亲的 \(2^{j-1}\) 级父亲”( $ 2^j=2\times 2^{j-1}$ )

    • 预处理

      for(int i = 1;i<=n;i++) f[i][0] = fa[i];
      for(int j = 1;j<=20;j++){
          for(int i = 1;i<=n;i++){
              f[i][j] = f[f[i][j-1]][j-1];
          }
      }
      
    • \(\operatorname{lca}\)

      1. \(u\)\(v\) 中较深的向上跳 \(|\text{dep}_u-\text{dep}_v|\)步,使他们的深度相同

      2. 如果此时 \(u\)\(v\) 相等则返回

      3. 不断从大到小枚举 \(i\) ,判断 \(f(u,i)\)\(f(v,i)\) 是否相同,若相同就跳过头了,若不同则同时向上跳 \(2^i\)

        当然也可以从小到大枚举 \(i\),寻找第一个可以使 \(f(u,i)=f(v,i)\)\(i\),同时向上跳 \(2^{i-1}\)

      4. \(f(u,0)=f(v,0)\)时,答案即为 \(f(u,0)\)\(f(v,0)\)

    • 代码

      int lca(int x,int y){
          if(dep[x] < dep[y]) swap(x,y);//令x为较深的点
          for(int i = 20;i>=0;i--){
              if(dep[f[x][i]] >= dep[y]){
                  x = f[x][i];//如果跳完深度仍然深于另一个点,则向上跳2^i级
              }
          }
          if(x == y) return x;//如果到达同一深度后两个点重合,则结果为较浅的点
          
          for(int i = 20;i>=0;i--){
              if(f[x][i] != f[y][i]){
                  x = f[x][i];
                  y = f[y][i];
                  //将两个点同时向上跳2^i步
              }
          }
          return f[x][0];
          
      }
      

树上dp

所谓树上dp,就是在树上进行的dp(经典废话

  • P1122 最大子树和

    给一颗树,每个点有点权,选择一个连通块使得点权和最大

    很明显这棵树中可能存在负权点,不然整棵树的权值和就是最大子树和

    \(f(i)\) 为以 \(i\) 为根的连通块的最大子权和

    易知 \(f(i)=f(i)+\max(f(j),0)\) ,其中 \(j\)\(i\) 的儿子

    void dfs(int fa,int x){
        for(int i = head[x];i;i = edge[i].next){
            int v = edge[i].to;
            if(v == fa) continue;
            dfs(x,v);
            f[x] += max(f[v],0);
        }
        ans = max(ans,f[x]);
    }
    
  • P1352 没有上司的舞会

    给一棵树,要求选择一些点,满足任意点和他的父亲不同时被选的情况下点权和最大

    \(f(i,0)\) 为 以 \(i\) 为根的子树且不选 \(i\) 的最大点权和,

    \(f(i,1)\) 为 以 \(i\) 为根的子树且 \(i\) 的最大点权和,

    那么可以推出,

    \(f(i,0)=f(i,0)+\max(f(j,0),f(j,1))\),其中 \(j\)\(i\) 的所有儿子

    \(f(i,1)=f(i,1)+f(j,0)\),其中 \(j\)\(i\) 的所有儿子

    \(\text{ans}\gets \max(f(1,0),f(1,1))\)

    void dfs(int fa,int x){
        f[x][1] = val[x];
        for(int i = head[x];i;i=edge[i].next){
            int v = edge[i].to;
            if(v == fa) continue;
            dfs(x,v);
            f[x][0] += max(f[v][1],f[v][0]);
            f[x][1] += f[v][0];
        }
    }
    int ans = max(f[1][0],f[1][1]);
    
  • P2458 保安站岗

    给定一棵树,要求在一些点上放置保安,保安可以看守该点即与该点相连的所有点,第 \(i\) 个点放置保安有代价 \(v_i\) ,求所有点都被看守的情况下的最小代价和

    \(f(i,0)\) 为以 \(i\) 为根的子树,\(i\)\(i\) 的父亲看守的最小代价和

    \(f(i,1)\) 为以 \(i\) 为根的子树,\(i\)\(i\) 他自己看守的最小代价和

    \(f(i,2)\) 为以 \(i\) 为根的子树,\(i\)\(i\) 的儿子看守的最小代价和

    1. 若一个点被其父亲看守,则该点无法看守其儿子节点,其儿子节点需要被【儿子自己】或【儿子的儿子】看守

      \(f(i,0)=f(i,0)+\min(f(j,1),f(j,2))\),其中 \(j\)\(i\) 的儿子

    2. 若一个点被自己看守,则该点的儿子可以被【儿子自己】,【该点本身】,【儿子的儿子】看守

      \(f(i,1)=f(i,1)+\min(f(j,0),f(j,1),f(j,2))\)

    3. 若一个点被其儿子看守,则其儿子节点可以被【儿子自己】,【儿子的儿子】看守,

      \(f(i,2)=f(i,2)+\min(f(j,1),f(j,2))\)

      但是如果转移方程仅仅是这样的话,当前节点可能没有人看守(即所有的儿子都由儿子的儿子看守时)

      所以至少要有一个儿子是自己看守自己

      不过如果出现这种情况,就表明所有的 \(\min(f(j,1),f(j,2))\) 都是 \(f(j,2)\) 更小一些

      这时可以定义一个 \(\text{tmp}\) 变量来记录 \(f(i,1)\)\(f(i,2)\) 的差值,即定义 \(\text{tmp}=f(i,1)-f(i,2)\)

      \(\text{tmp} < 0\) 则说明 \(f(i,1) < f(i,2)\) ,即出现了一个自己看管自己的儿子节点,就无须理会了

      \(\text{tmp} > 0\) 则说明当前的儿子节点是由儿子的儿子看管的,可以被修正为自己看管

      而且当 \(\text{tmp}\) 为正且越小的时候,说明将该儿子节点修改为自己看管亏损得越少

    所以可以写出如下的代码

    void dfs(int fa,int x){
        f[x][1] = val[x];
        int tmp = 1e9;
        for(int i = head[x];i;i=edge[i].next){
            int v = edge[i].to;
            if(v == fa) continue;
            tmp = min(tmp,f[v][1]-f[v][2]);
            f[x][0] += min(f[v][1],f[v][2]);
            f[x][1] += min(min(f[v][0],f[v][1]),f[v][2]);
            f[x][2] += min(f[v][1],f[v][2]);
        }
        f[x][2] += max(tmp,0);
        //若tmp为正数则加上tmp起到修改作用
        //f[j][2]+(f[j][1]-f[j][2])=f[j][1]
    }
    int ans = max(f[1][1],f[1][2]);
    

其他

又是亿些概念

  1. 树的重心:选一个点为根,使最大子树最小的点称为树的重心
  2. 树的直径:树上最长路径
  3. 树的中点:直径中点(树的中心不等于树的中心)

又是亿些性质

  1. 树直径可能有很多条,他们必定交于树的中心
  2. 树的中心是一个或两个
  3. 以树的重心为根是,所有子树的大小都不超过整棵树大小的一半
  4. 树的重心是一个或两个
  5. 两树相连,重心在原来两树重心的路径上
posted @ 2022-02-14 20:02  Burnling  阅读(44)  评论(0编辑  收藏  举报