四叉树交集

夏虫语火

已于 2022-07-16 13:00:07 修改

阅读量82

点赞数

分类专栏： leetcode题解文章标签： leetcode 算法 c++

于 2022-07-16 12:47:55 首次发布

本文链接：https://blog.csdn.net/weixin_45713539/article/details/125802520

版权

leetcode题解专栏收录该内容

2 篇文章 0 订阅

订阅专栏

558. 四叉树交集

解题思路

递归，我是在 $q u a d T ree 1$ 的基础上直接修改的，题目很长可能理解有点困难

实际上只要考虑叶子节点的值，以及它是否会影响父亲节点成为叶子节点（题目中说明了这种情况）

设 $t 、 p$ 分别为当前节点

都是叶子节点直接计算或的结果赋给 $t$ 它是要改变的树
$t$ 是叶子节点考虑它的 $v a l$
1. $v a l$ =1 直接返回，1 | 任意数都是1
2. $v a l$ =0 把 $p$ 节点赋给 $s$ ，0 | 任意数都是任意数本身
$s$ 是叶子节点同理2

坑点： $t = p$ 直接赋值不行

代码

相比官方题解 新建Node不会修改原四叉树，而且更好看一点
Node* intersect(Node* quadTree1, Node* quadTree2) {
    Node* t = quadTree1;
    Node* p = quadTree2;
    solve(t, p);
    return quadTree1;
}
void solve(Node* t, Node* p)
{
    if(t->isLeaf&&p->isLeaf)
    {
        t->val = t->val|p->val;
        p->val = t->val;
        return;
    } else if(t->isLeaf){
        if(!t->val)
        {
            t->val = p->val;
            t->isLeaf = p->isLeaf;
            t->topLeft = p->topLeft;
            t->topRight = p->topRight;
            t->bottomLeft = p->bottomLeft;
            t->bottomRight = p->bottomRight;
        }
        return;
    } else if(p->isLeaf) {
        if(p->val)
        {
            t->val = p->val;
            t->isLeaf = p->isLeaf;
            t->topLeft = p->topLeft;
            t->topRight = p->topRight;
            t->bottomLeft = p->bottomLeft;
            t->bottomRight = p->bottomRight;
        }
        return;
    }
    solve(t->topLeft, p->topLeft);
    solve(t->topRight, p->topRight);
    solve(t->bottomLeft, p->bottomLeft);
    solve(t->bottomRight, p->bottomRight);
    if(t->topLeft->isLeaf&&t->topRight->isLeaf&&t->bottomLeft->isLeaf&&t->bottomRight->isLeaf&&t->topLeft->val==t->topRight->val&&t->topRight->val==t->bottomLeft->val&&t->bottomLeft->val==t->bottomRight->val)
    {
        t->isLeaf=true;
        t->val = t->topLeft->val;
        t->topLeft = NULL;
        t->topRight = NULL;
        t->bottomLeft = NULL;
        t->bottomRight = NULL;
    }
}