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

思路一是在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);
    }

};