LeetCode——二叉树的堂兄弟节点

题目描述：

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

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

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

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

代码如下：

/**
 * 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) {
        queue<TreeNode*>q;
        q.push(root); 
        int flag=2;                      
        while(!q.empty())
        {
            int n=q.size();
            for(int i=0;i<n;i++)
            {         
                root=q.front();       
                q.pop();    
                if(root->left&&root->right&&((root->left->val==x&&root->right->val==y)||(root->left->val==y&&root->right->val==x)))                 
                {           
                    return false;        
                } 
                if(root->left)
                {       
                    q.push(root->left);    
                    if(root->left->val==x||root->left->val==y) 
                        flag--;    
                }
                if(root->right)
                {      
                    q.push(root->right);
                    if(root->right->val==x||root->right->val==y) 
                        flag--;
                }  
            }
            if(flag==1)
            {
                return false;  
            } 
            if(flag==0)
            { 
                return true;
            };    
        }                              
        return false;  
    }
};

执行结果：