LeetCode刷题实战558：四叉树交集-优快云博客

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选！

今天和大家聊的问题叫做 四叉树交集，我们先来看题面：

https://leetcode-cn.com/problems/logical-or-of-two-binary-grids-represented-as-quad-trees/

示例

示例 2：
输入：quadTree1 = [[1,0]]
, quadTree2 = [[1,0]]
输出：[[1,0]]
解释：两个数所表示的矩阵大小都为 1*1，值全为 0 
结果矩阵大小为 1*1，值全为 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]]

解题

https://leetcode-cn.com/problems/logical-or-of-two-binary-grids-represented-as-quad-trees/solution/zhe-ti-you-ji-ge-xu-yao-zhu-yi-de-di-fang-yi-zhi-k/

A || B 的意思是指

如果A为叶子节点，并且A的值等于true, 这时直接返回true, 不用管B

如果A为叶子节点，但是A的值等于false, 这时直接返回B就可以了（*注意即使B不是一个叶子节点，也要直接返回，可以参照后面的图片）

如果A不为叶子节点，原理一样，这时看B，如果B为叶子节点，并且B的值为true, 则直接返回true, 不用管A

如果A不为叶子节点， B为叶子节点，并且B的值为false, 这时直接返回A

如果A和B都不为叶子节点，这时分别再讲A, B的四个子节点传入下一次递归计算结果

最后，如果A和B的四个子节点的值都会true的话，这时需要进行合并为一个节点。

class Solution {
public:
  Node* intersect(Node* quadTree1, Node* quadTree2) {

    if (quadTree1->isLeaf) { // 1.如果A是叶子节点
      if (quadTree1->val) { // 2.如果A的值为true则直接返回A树
        return quadTree1;
      }
      else { 
        return quadTree2; // 3.A为叶子但A的值为false, 则返回B树
      }
    }
    if (quadTree2->isLeaf) { // 4.如果B是叶子节点
      if (quadTree2->val) { // 5.如果B的值为true则直接返回B树
        return quadTree2;
      }
      else {
        return quadTree1; // 6.B为叶子节点，但B的值为false，则返回A树
      }
    }
    // 7.A和B都不为叶子节点的情况下，分别递归获取四个方向的子节点
    Node* topLeft = intersect(quadTree1->topLeft, quadTree2->topLeft);
    Node* topRight = intersect(quadTree1->topRight, quadTree2->topRight);
    Node* bottomLeft = intersect(quadTree1->bottomLeft, quadTree2->bottomLeft);
    Node* bottomRight = intersect(quadTree1->bottomRight, quadTree2->bottomRight);
    
    // 8.如果四个节点同时为叶子节点。并且值为true时,则返回的节点进行合并,合并为一个值为true的叶子节点
    if (topLeft->isLeaf && topRight->isLeaf && bottomLeft->isLeaf && bottomRight->isLeaf &&
      topLeft->val && topRight->val && bottomLeft->val && bottomRight->val) {
      return new Node(true, true, NULL, NULL, NULL, NULL);
    }
    // 9. 通过第7步获取得四个方位的节点，构造出一个新的节点树
    return new Node(false, false, topLeft, topRight, bottomLeft, bottomRight);
  }

};

好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力。

上期推文：

LeetCode1-540题汇总，希望对你有点帮助！

LeetCode刷题实战541：反转字符串 II

LeetCode刷题实战542：01 矩阵

LeetCode刷题实战543：二叉树的直径

LeetCode刷题实战544：输出比赛匹配对

LeetCode刷题实战545：二叉树的边界

LeetCode刷题实战546：移除盒子

LeetCode刷题实战547：省份数量

LeetCode刷题实战548：将数组分割成和相等的子数组

LeetCode刷题实战549：二叉树中最长的连续序列