LeetCode993. 二叉树的堂兄弟节点

题目

class Solution {
public:
    TreeNode* r1;TreeNode* r2;
    bool isCousins(TreeNode* root, int x, int y) {
        dfs(root,r1,x);
        dfs(root,r2,y);
        int h_x = Depth(root,x);
        int h_y = Depth(root,y);
        if(h_x == h_y && r1 != r2) return true;
        return false;
    }
    void dfs(TreeNode* root,TreeNode* &r,int v){
        if(root!= NULL) {
            dfs(root->left,r,v);
            if(root->left &&root->left->val == v ) r = root;
            if(root->right && root->right->val == v) r = root;
            dfs(root->right,r,v);
        }
    }
    int Depth(TreeNode* root,int v){
        if(root == NULL) return -1;
        if(root->val == v) return 0;
        //TreeNode* node;
        queue<TreeNode*>q;
        q.push(root);
        int depth = 0;
        while(!q.empty()){
            int num = q.size(); //一定要有这句，否则会影响下面循环次数
            for(int i = 0;i < num;i++){
                TreeNode* node = q.front();q.pop();
                if(node->val == v) return depth;
                if(node->left != NULL) q.push(node->left);
                if(node->right != NULL) q.push(node->right);
            }
            depth++;
        }
        return depth;
    }
};

注意：

1.依旧是函数的形参为指针情况，如果想要保存对指针的修改需要址传递 如dfs中 TreeNode* &r

2.第28行代码别忘记，一定要提前记录每一层的个数