蓝桥杯 之 LCA算法

文章目录



习题

1483.树节点的第K个祖先

1483.树节点的第K个祖先

  • 普通的做法,当然是一个个往上面搜索,但是这样的话时间复杂度是o(k),那么能不能每次求解的是爷爷节点,这样就是按照二进制的步子进行寻找



python 复制代码
class TreeAncestor:

    def __init__(self, n: int, parent: List[int]):
        # bit_length()表示二进制的位数,因为pa[x][0]直接就是父亲节点,所以只用少一位即可
        m = n.bit_length() - 1
        # 初始化
        pa = [[p] + [-1]*m for p in parent]
        # 注意这个遍历,先枚举这个i再枚举这个x,先算出全部节点的所有爷爷节点,再算出所有爷爷的爷爷节点
        for i in  range(m):
            for x in range(n):
                p = pa[x][i]
                if p != -1:
                    pa[x][i+1] = pa[p][i]
        self.pa = pa

    def getKthAncestor(self, node: int, k: int) -> int:
        for i in range(k.bit_length()):
            # 这个k>>i的判断就十分巧妙
            # 并且这个node 是全局的,i从0开始判断
            if (k >> i) & 1:
                node = self.pa[node][i]
                if node < 0 :
                    break
        return node

拓展:LCA


  • 初始化
python 复制代码
# 
python 复制代码
class TreeAncestor:
    def __init__(self, edges: List[List[int]]):
        n = len(edges) + 1
        m = n.bit_length()
        g = [[] for _ in range(n)]
        for x, y in edges:  # 节点编号从 0 开始
            g[x].append(y)
            g[y].append(x)

        depth = [0] * n
        pa = [[-1] * m for _ in range(n)]
        def dfs(x: int, fa: int) -> None:
            pa[x][0] = fa
            for y in g[x]:
                if y != fa:
                    depth[y] = depth[x] + 1
                    dfs(y, x)
        dfs(0, -1)

        for i in range(m - 1):
            for x in range(n):
                if (p := pa[x][i]) != -1:
                    pa[x][i + 1] = pa[p][i]
        self.depth = depth
        self.pa = pa

    def get_kth_ancestor(self, node: int, k: int) -> int:
        for i in range(k.bit_length()):
            if (k >> i) & 1:  # k 二进制从低到高第 i 位是 1
                node = self.pa[node][i]
        return node

    # 返回 x 和 y 的最近公共祖先(节点编号从 0 开始)
    def get_lca(self, x: int, y: int) -> int:
        if self.depth[x] > self.depth[y]:
            x, y = y, x
        # 使 y 和 x 在同一深度
        y = self.get_kth_ancestor(y, self.depth[y] - self.depth[x])
        if y == x:
            return x
        for i in range(len(self.pa[x]) - 1, -1, -1):
            px, py = self.pa[x][i], self.pa[y][i]
            if px != py:
                x, y = px, py  # 同时上跳 2**i 步
        return self.pa[x][0]
相关推荐
不吃元西几秒前
leetcode 118. 杨辉三角
算法·leetcode·职场和发展
脑子慢且灵3 分钟前
【蓝桥杯】 枚举和模拟练习题
java·开发语言·职场和发展·蓝桥杯·模拟·枚举
wen__xvn7 分钟前
每日一题洛谷P8716 [蓝桥杯 2020 省 AB2] 回文日期c++
c++·算法·蓝桥杯
梭七y11 分钟前
【力扣hot100题】(027)两数相加
算法·leetcode·职场和发展
破东风15 分钟前
leetcode每日一题:向字符串添加空格
算法·leetcode·模拟
rigidwill66617 分钟前
LeetCode hot 100—二叉搜索树中第K小的元素
数据结构·c++·算法·leetcode·职场和发展
银河小铁骑plus21 分钟前
力扣hot100_贪心算法
算法·leetcode·贪心算法
小郝 小郝24 分钟前
(C语言)指针运算 习题练习1.2(压轴难题)
java·开发语言·算法
welkin37 分钟前
算法区间合并问题
前端·算法
星途码客1 小时前
C++位运算精要:高效解题的利器
java·c++·算法