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

题目描述
在二叉树中，根节点位于深度 0 处，每个深度为 k 的节点的子节点位于深度 k+1 处。

• 如果二叉树的两个节点深度相同，但 父节点不同 ，则它们是一对堂兄弟节点。
• 我们给出了具有唯一值的二叉树的根节点 root ，以及树中两个不同节点的值 x 和 y 。
• 只有与值 x 和 y 对应的节点是堂兄弟节点时，才返回 true 。否则，返回 false。

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

基本思路：找到给定的两个节点的父节点以及两个节点所处的深度。
可以在递归的过程中同时遍历出来，因为递归遍历会把整棵树遍历完，只要遍历过程中某一节点的值与x或者y相等即可将该节点所处的深度以及所属父节点记录下来。那么如果一次性记录的话，dfs传入的参数除了题目给定的root，x,y,还有深度depth以及当前节点的根节点。只要两个节点都找到了便可以跳出递归。深度depth这个参数是随着递归的调用不断变化的，只要进入一层递归深度就要加一。

class Solution {
public:
    int lenx,leny;
    TreeNode* rootx;
    TreeNode* rooty;
    bool findx=false,findy=false;
    bool isCousins(TreeNode* root, int x, int y) {
        dfs(root,0,x,y,root);
        return rootx!=rooty && lenx == leny;
    }
    void dfs(TreeNode* root,int depth,int x,int y,TreeNode* node)
    {
        if(!root) return;
        if(root->val == x ) 
        {
            rootx=node;
            findx=true;
            lenx=depth;
        }
        if(root->val == y)
        {
            rooty=node;
            findy=true;
            leny=depth;
        }
        if(findx&&findy) return;
        dfs(root->left,depth+1,x,y,root);
        dfs(root->right,depth+1,x,y,root);
    }
};