leetcode——第993题——二叉树的堂兄弟节点

题目：
在二叉树中，根节点位于深度 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 {
/*
本题使用层序遍历，也就是广度优先遍历，也即是BST是最佳方案
但是我第一次在自己考虑的时候，仅仅考虑到需要再同一层，没有考虑到不能是同一个父亲
因此 看了题解，发现同一个父亲的标记也很简单
只需要在队列中插入元素的时候，将父节点也插入，然后就可以对应的判断啦~~

注意pair元素的赋值需要用 大括号
*/

public:
    bool isCousins(TreeNode* root, int x, int y) 
    {
        // 方法一：队列里存放两个树节点，一个是 cur,一个是 parent
        // 注意这里是que里是 pair 类型，这样的类型，赋值操作都要用大括号 
        queue<pair<TreeNode*, TreeNode*>> que;
        if(root == nullptr) return false;
        // 顺序不要反了，当前节点  +  父节点
        que.push({root, nullptr});
        while(!que.empty())
        {
            int size = que.size();      // 更新每一层的 size
            vector<TreeNode*> pare; // 记录两次的父节点
            for(int i = 0; i < size; ++i)
            {
                auto[cur, parent] = que.front();
                que.pop();  // 这里不要忘记 pop尾删~~~
                if(cur->val == x || cur->val == y)  pare.push_back(parent);

                // 既要在同一层，又要不是同一个父亲
                if(pare.size() == 2 && pare[0] != pare[1])  return true;

                if(cur->left != nullptr)   que.push({cur->left, cur});//push的顺序不要反了
                if(cur->right != nullptr)   que.push({cur->right, cur});
            }
            if(pare.size() == 1)    return false;
        }
        return false;
    }
};