(刷题笔记) Leetcode 993. 二叉树的堂兄弟节点

题目

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

如果二叉树的两个节点深度相同，但父节点不同，则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root，以及树中两个不同节点的值 x 和 y。

只有与值 x 和 y 对应的节点是堂兄弟节点时，才返回 true。否则，返回 false。

示例 1：

输入：root = [1,2,3,4], x = 4, y = 3
输出：false

示例 2：

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

示例 3：

输入：root = [1,2,3,null,4], x = 2, y = 3
输出：false

提示：

二叉树的节点数介于 2 到 100 之间。
每个节点的值都是唯一的、范围为 1 到 100 的整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cousins-in-binary-tree

解题思路

说实话还是第一次写层序遍历，不会写，看了一下解答区，发现是用队列实现的。

前面写中序遍历这种都是用栈实现的，现在开始用队列了还不太习惯。

不过写二叉树这一块的题目好处就是能把栈队列树都学一遍。

这一题的思路就是判断同一层中是否有满足分别等于x和y的，然后排除掉是同一父节点的情况，剩下的就可以下结论是否是堂兄弟节点。

代码(C++)

/**
 * 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 {
public:
    bool isCousins(TreeNode* root, int x, int y) {

        if(root==NULL) return false;
        queue<TreeNode*> q;
        q.push(root);
        if(root->val==x||root->val==y) return false;

        while(!q.empty()){

            int qsz=q.size();
            bool isx=false;
            bool isy=false;
            for(int i=0;i<qsz;++i){

                TreeNode* node=q.front();
                q.pop();

                if(node->val==x) isx=true;
                if(node->val==y) isy=true;
                
                if(node->left&&node->right){
                    if(node->left->val==x&&node->right->val==y){
                        return false;
                    }
                    if(node->left->val==y&&node->right->val==x){
                        return false;
                    }

                }
                
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);

            }

             if(isx&&isy) return true;


        }

        return false;
        
        
    }
};