二叉树的堂兄弟节点

ZXY-YXZ

于 2021-08-25 21:26:09 发布

阅读量55

点赞数

分类专栏：实习刷题文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/weixin_47791283/article/details/119919882

版权

实习刷题专栏收录该内容

177 篇文章 0 订阅

订阅专栏

993. 二叉树的堂兄弟节点 - 力扣（LeetCode） (leetcode-cn.com)

这是我见过最最最简洁的代码。。。。。。原来求深度可以这么求。。。。

class Solution {
    int xf,xd,yf,yd;
public:
    bool isCousins(TreeNode* root, int x, int y) {
       dfs(root->left,1,x,y,root->val);
       dfs(root->right,1,x,y,root->val);
       return (xf!=yf)&&(xd==yd);
    }

    void dfs(TreeNode*root,int d,int x,int y,int vf){
       if(root==nullptr) return;
       if(root->val==x){
           xf=vf;
           xd=d;
       }
       else if(root->val==y){
           yf=vf;
           yd=d;
       }
       else{
            dfs(root->left,d+1,x,y,root->val);
            dfs(root->right,d+1,x,y,root->val);

       }
    }
};