Leetcode 993. 二叉树的堂兄弟节点（DFS记录节点信息）

最新推荐文章于 2025-03-30 03:15:00 发布

wwxy261

最新推荐文章于 2025-03-30 03:15:00 发布

阅读量142

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/116905822

版权

算法专栏收录该内容

3633 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)的方法来确定二叉树中的两个节点是否为堂兄弟节点。通过记录每个目标节点的父节点及深度，可以有效地判断它们是否处于同一深度且不共用同一个父节点。

摘要由CSDN通过智能技术生成

思路一是在BFS搜索同一层的过程中判断，这种写法不是很优雅。

这里采用一个简单直接的做法，直接在DFS的时候，记录x和y节点的父亲节点，以及深度参数，然后直接判断即可。

/**
 * 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 {

private:
    // x 的信息
    int x;
    TreeNode* x_parent;
    int x_depth;
    bool x_found = false;

    // y 的信息
    int y;
    TreeNode* y_parent;
    int y_depth;
    bool y_found = false;

public:
    bool isCousins(TreeNode* root, int x_, int y_) {
        x = x_;
        y = y_;
        dfs(root, nullptr, 0);
        return x_found && y_found &&  x_depth == y_depth && x_parent != y_parent;
    }

    void dfs(TreeNode* root, TreeNode* parent, int depth){
        if(root == nullptr){
            return;
        }
        if(root->val == x){
            x_parent = parent;
            x_depth = depth;
            x_found = true;
        }
        if(root->val == y){
            y_parent = parent;
            y_depth = depth;
            y_found = true;
        }
        dfs(root->left, root, depth + 1);
        dfs(root->right, root, depth + 1);
    }

};