leetcode_993. 二叉树的堂兄弟节点（C++，广搜）

题目

原题链接

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

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

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

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

思路

首先思考，堂兄弟节点要满足什么条件呢？

• 两个结点的爹不同
• 两个结点在一层

看到“在同一层”，你的第一反应是什么？广搜！

广搜时，遵循以下规则：

1. 先看当前节点的孩子是不是x、y，是就返回false， 提前结束， 缩短时间
2. 按层遍历，当发现一层中含有x、y两个结点，返回true， 因为在此之前，已经判断过第一点了

思路已经很清晰了，啥都别说了，开门，放狗代码！

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isCousins(TreeNode* root, int x, int y) {
        queue<TreeNode*> q;
        bool fx = false, fy = false; //这是两个标记变量，代表在这一层中，是否发现了x和y
        q.push(root); //先将根节点入队
        while(!q.empty()) { //当队列不为空
            fx = fy = false; //记得每轮循环一开始都要初始化
            int len = q.size(); //维护长度
            for(int i = 0; i < len; ++ i) {
                TreeNode* t = q.front(); q.pop();
                if(t -> left && t -> right)
                    if((t->left->val==x && t->right->val==y) || (t->left->val==y && t->right->val==x))
                        return false; //对应着上面的第一点
                if(t -> val == x) fx = true; //发现x则标记
                if(t -> val == y) fy = true; //发现y则标记
                if(fx && fy) return true; //x和y都在同一层找到，返回true
                if(t -> left) q.push(t -> left); //左儿子入队
                if(t -> right) q.push(t -> right); //右儿子入队
            }
        }
        return false; //找不到，返回false
    }
};