力扣993-二叉树的堂兄弟节点

二叉树的堂兄弟节点

题目链接

解题思路

  1. 题目要求我们找到x和y处于同样的深度,但是父节点不同,返回true,否则返回false
  2. dfs()函数参数解读
    1. TreeNode*root:是此时正要处理的节点
    2. int x:表示我们要寻找的x和y
    3. int fa:表示正在处理的节点的父节点的值
    4. int depth:表示此时正在处理的节点的深度
  3. dfs()函数返回一个数组,第一个值是目标节点的父节点的值,第二个值是目标节点的深度
  4. 如果说当前处理的节点为空,那么直接返回{0,0}
  5. 如果找到目标节点,则返回父节点和深度{fa,depth}
  6. 之后就是递归左子树和右子树,深度+1,父节点为当前节点的值
  7. 其实,左右子树递归时,只有一个子树包含目标值(因为所有的值只会出现一次),另一个子树返回{0,0}
  8. 所以最后返回{l[0]+r[0],l[1]+r[1]};
c++ 复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> dfs(TreeNode* root,int x,int fa,int depth){
        if(!root) return {0,0};
        if(root->val == x) return {fa,depth};
        auto l = dfs(root->left,x,root->val,depth+1);
        auto r = dfs(root->right,x,root->val,depth+1);
        return {l[0]+r[0],l[1]+r[1]};
    }
    bool isCousins(TreeNode* root, int x, int y) {
        auto a = dfs(root,x,-1,0);
        auto b = dfs(root,y,-1,0);
        if(a[0] != b[0] && a[1] ==b[1]){
            return true;
        }else{
            return false;
        }
    }
};
相关推荐
我爱C编程1 小时前
基于Qlearning强化学习的1DoF机械臂运动控制系统matlab仿真
算法
chao_7891 小时前
CSS表达式——下篇【selenium】
css·python·selenium·算法
chao_7892 小时前
Selenium 自动化实战技巧【selenium】
自动化测试·selenium·算法·自动化
YuTaoShao2 小时前
【LeetCode 热题 100】24. 两两交换链表中的节点——(解法一)迭代+哨兵
java·算法·leetcode·链表
怀旧,2 小时前
【数据结构】8. 二叉树
c语言·数据结构·算法
泛舟起晶浪2 小时前
相对成功与相对失败--dp
算法·动态规划·图论
地平线开发者2 小时前
地平线走进武汉理工,共建智能驾驶繁荣生态
算法·自动驾驶
IRevers3 小时前
【自动驾驶】经典LSS算法解析——深度估计
人工智能·python·深度学习·算法·机器学习·自动驾驶
前端拿破轮3 小时前
翻转字符串里的单词,难点不是翻转,而是正则表达式?💩💩💩
算法·leetcode·面试
凤年徐3 小时前
【数据结构与算法】203.移除链表元素(LeetCode)图文详解
c语言·开发语言·数据结构·算法·leetcode·链表·刷题