目录

题目描述:

在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

如果二叉树的两个节点深度相同,但父节点不同,则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root,以及树中两个不同节点的值 xy

只有与值 xy 对应的节点是堂兄弟节点时,才返回 true。否则,返回 false

示例 1:

leetcode 993. 二叉树的堂兄弟节点(Cousins in Binary Tree) 随笔 第1张

输入:root = [1,2,3,4], x = 4, y = 3
输出:false

示例 2:

leetcode 993. 二叉树的堂兄弟节点(Cousins in Binary Tree) 随笔 第2张

输入:root = [1,2,3,null,4,null,5], x = 5, y = 4
输出:true

示例 3:

leetcode 993. 二叉树的堂兄弟节点(Cousins in Binary Tree) 随笔 第3张

输入:root = [1,2,3,null,4], x = 2, y = 3
输出:false

提示:

  1. 二叉树的节点数介于 2100 之间。
  2. 每个节点的值都是唯一的、范围为 1100 的整数。

解法:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool getPath(TreeNode* root, int x, vector<int>& path){
        if(root == NULL){
            return false;
        }else {
            path.push_back(root->val);
            if(root->val != x){
                if(getPath(root->left, x, path)){
                    return true;
                }
                if(getPath(root->right, x, path)){
                    return true;
                }
                path.pop_back();
                return false;
            }else{
                return true;
            }
        }
    }
    
    bool isCousins(TreeNode* root, int x, int y) {
        vector<int> path1, path2;
        getPath(root, x, path1);
        getPath(root, y, path2);
        if(path1.size() != path2.size()){
            return false;
        }else{
            path1.pop_back();
            path2.pop_back();
            return path1.back() != path2.back();
        }
    }
};
扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄