当前位置:   article > 正文

⭐算法入门⭐《二叉树》简单07 —— LeetCode 993. 二叉树的堂兄弟节点_root = [1,2,3,4], x = 4, y = 3

root = [1,2,3,4], x = 4, y = 3

一、题目

1、题目描述

  在二叉树中,根节点位于深度 0 0 0 处,每个深度为 k k k 的节点的子节点位于深度 k + 1 k+1 k+1 处。如果二叉树的两个节点深度相同,但父节点不同 ,则它们是一对堂兄弟节点。我们给出了具有唯一值的二叉树的根节点 r o o t root root ,以及树中两个不同节点的值 x x x y y y。只有与值 x x x y y y 对应的节点是堂兄弟节点时,才返回 true。否则,返回 false
  样例输入: root = [1,2,3,4], x = 4, y = 3
  样例输出: false

2、基础框架

  • C语言 版本给出的基础框架代码如下:
bool isCousins(struct TreeNode* root, int x, int y){
}
  • 1
  • 2

3、原题链接

LeetCode 993. 二叉树的堂兄弟节点

二、解题报告

1、思路分析

  递归将每个结点的深度信息和父结点信息保存下来;然后对传进来的 x x x y y y 进行条件判定;

2、时间复杂度

  由于需要把所有结点都遍历一遍,所以时间复杂度为 O ( n ) O(n) O(n)

3、代码详解

#define maxn 110
#define nullparent -102001021

int depth[maxn];
int parent[maxn];

void getInfo(struct TreeNode* root, int p, int d) {   // (1)
    if(root == NULL) {
        return ;
    }
    parent[ root->val ] = p;                          // (2)
    depth[ root->val ] = d;                           // (3)
    getInfo(root->left, root->val, d + 1);            // (4)
    getInfo(root->right, root->val, d + 1);           // (5)
}

bool isCousins(struct TreeNode* root, int x, int y){
    if(x == y) {
        return false;                                 // (6)
    }
    getInfo(root, nullparent, 0);                     // (7)
    if(depth[x] == depth[y]) {
        if(parent[x] != parent[y]) {                
            return true;                              // (8)
        }
    }
    return false;                                     // (9)
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • ( 1 ) (1) (1) 递归将每个结点的深度信息和父结点信息保存下来;
  • ( 2 ) (2) (2) 保存当前结点的父结点信息;
  • ( 3 ) (3) (3) 保存当前结点的深度信息;
  • ( 4 ) (4) (4) 递归计算左子树;
  • ( 5 ) (5) (5) 递归计算右子树;
  • ( 6 ) (6) (6) 如果是同一个结点,肯定不是堂兄弟;
  • ( 7 ) (7) (7) 计算每个结点的父结点和深度;
  • ( 8 ) (8) (8) 满足堂兄弟条件,返回 true
  • ( 9 ) (9) (9) 不满足条件,返回false

三、本题小知识

  递归时处理二叉树问题很好的手段。


四、加群须知

  相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」
  那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:

推荐阅读
相关标签