【从零开始的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 的整数。

我的解答

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private int height = 0; // 记录x，y所在节点的深度，一致时为-1,不一致时为-2
    private int parent = 0; // 记录x，y的父节点，不一致时为-1，一致时为-2
    private boolean isFind = false;

    public boolean isCousins(TreeNode root, int x, int y) {
        
        return  findXY(root,1,x,y);
    }

    private boolean findXY(TreeNode root,int h,int x ,int y){
        // 已经确认是堂兄弟节点了，就不用再递归遍历其他节点了
        if(isFind) return true;
        h++;
        if(root.left != null) {
            // 如果找到了x或y，就记录它的高度和父节点，且不再往下找
            if(root.left.val == x || root.left.val == y){
                if(height == 0) {
                    height = h;
                    parent = root.val;
                }
                // 是堂兄弟节点
                else if(height == h && parent != root.val) return isFind = true;
                return false;
            }
            // 遍历左节点
            findXY(root.left,h,x,y);
        }
        if(root.right != null) {
            // 如果找到了x或y，就记录它的高度和父节点，且不再往下找
            if(root.right.val == x || root.right.val == y){
                if(height == 0) {
                    height = h;
                    parent = root.val;
                }
                // 是堂兄弟节点
                else if(height == h && parent != root.val) return isFind = true;
                return false;
            }
            // 遍历右节点
            findXY(root.right,h,x,y);
        }
        return isFind;
    }
}