leetcode - 20 558. 四叉树交集

datawhale-leafy

于 2022-07-15 12:22:20 发布

阅读量103

点赞数

分类专栏： leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_42906799/article/details/125801831

版权

leetcode 专栏收录该内容

33 篇文章 0 订阅

订阅专栏

二进制矩阵中的所有元素不是 0 就是 1 。

给你两个四叉树，quadTree1 和 quadTree2。其中 quadTree1 表示一个 n * n 二进制矩阵，而
quadTree2 表示另一个 n * n 二进制矩阵。

请你返回一个表示 n * n 二进制矩阵的四叉树，它是 quadTree1 和 quadTree2 所表示的两个二进制矩阵进行
按位逻辑或运算的结果。

示例 2：

输入：quadTree1 = [[1,0]] , quadTree2 = [[1,0]] 输出：[[1,0]]
解释：两个数所表示的矩阵大小都为 11，值全为 0 结果矩阵大小为 11，值全为 0 。

示例 3：

输入：quadTree1 = [[0,0],[1,0],[1,0],[1,1],[1,1]] , quadTree2 =
[[0,0],[1,1],[1,1],[1,0],[1,1]] 输出：[[1,1]]

示例 4：

输入：quadTree1 = [[0,0],[1,1],[1,0],[1,1],[1,1]] , quadTree2 =
[[0,0],[1,1],[0,1],[1,1],[1,1],null,null,null,null,[1,1],[1,0],[1,0],[1,1]]
输出：[[0,0],[1,1],[0,1],[1,1],[1,1],null,null,null,null,[1,1],[1,0],[1,0],[1,1]]

示例 5：

输入：quadTree1 =
[[0,1],[1,0],[0,1],[1,1],[1,0],null,null,null,null,[1,0],[1,0],[1,1],[1,1]]
, quadTree2 = [[0,1],[0,1],[1,0],[1,1],[1,0],[1,0],[1,0],[1,1],[1,1]]
输出：[[0,0],[0,1],[0,1],[1,1],[1,0],[1,0],[1,0],[1,1],[1,1],[1,0],[1,0],[1,1],[1,1]]

我们可以按以下步骤为二维区域构建四叉树：

如果当前网格的值相同（即，全为 0 或者全为 1），将 isLeaf 设为 True ，将 val 设为网格相应的值，并将四个子节点都设为
Null 然后停止。如果当前网格的值不同，将 isLeaf 设为 False，将 val
设为任意值，然后如下图所示，将当前网格划分为四个子网格。使用适当的子网格递归每个子节点。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/logical-or-of-two-binary-grids-represented-as-quad-trees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

最简单的思路是首先将四叉树转换为矩阵再做按位或的运算，但这样的问题是，完全没有利用到按位或运算和四叉树，因此是极为低效的方法。

我们注意到在按位或运算中，1|x=1, 0|x=x，因此我们在运算时并不一定需要具体到一位一位运算，只需要保证两边对应的分块大小一致时，利用叶子节点元素相同的性质进行分块运算。因此可以按叶子节点计算。自然也就引出了思路，递归遍历四叉树，到叶子节点就计算

class Solution:
    def intersect(self, quadTree1: 'Node', quadTree2: 'Node') -> 'Node':
        if quadTree1.isLeaf:
            return Node(True, True) if quadTree1.val else quadTree2
        if quadTree2.isLeaf:
            return self.intersect(quadTree2, quadTree1)
        o1 = self.intersect(quadTree1.topLeft, quadTree2.topLeft)
        o2 = self.intersect(quadTree1.topRight, quadTree2.topRight)
        o3 = self.intersect(quadTree1.bottomLeft, quadTree2.bottomLeft)
        o4 = self.intersect(quadTree1.bottomRight, quadTree2.bottomRight)
        if o1.isLeaf and o2.isLeaf and o3.isLeaf and o4.isLeaf and o1.val == o2.val == o3.val == o4.val:
            return Node(o1.val, True)
        return Node(False, False, o1, o2, o3, o4)