【LeetCode】第993题——二叉树的堂兄弟节点（难度：简单）

题目描述

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

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

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

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

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

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

3. 示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

就是简单的DFS即可，因为节点的值各不相同，因此记录一下父节点的值和x、y的深度即可。这里应该用HashMap，但是为了取值方便我这里使用了List，两个值为一对，分别为父节点的值和深度

另外有几点可以提前结束当前分支的递归

1. 如果树为空，或者根节点的值就与x或y相等，直接可以返回false，不用递归
2. 如果找到x或y的一个值，当前分支的递归可以直接终止，因为如果要返回true就不可能为父子节点的关系。也因此List的长度可能为2而非为4，最后如果List的长度为2则返回false
3. x和y都找到后可以直接终止递归，不用继续递归至全遍历

代码详解

/**
 * 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 {

    List<Integer> list =  new ArrayList<>(); // 当然也可以用Map
    int count = 2; // 表示x和y还有几个没被找到

    public boolean isCousins(TreeNode root, int x, int y) {
        if(root == null || root.val == x || root.val == y) {
            return false; // 递归前，有几种特殊情况可以直接返回false
        }
        dfs(root, x, y, 0); // 开始递归，最后一个参数为深度
        if(list.size() != 4) {
            return false; // 递归方案的设计，导致list长度可能不为4而为2，这时说明x与y是处于同一颗子树的不同深度上
        }
        if(list.get(0) != list.get(2) && list.get(1) == list.get(3)) {
            return true; // 唯一返回true的条件：父节点不同且深度相同
        }
        return false;
    }

    public void dfs(TreeNode node, int x, int y, int deep) {
        if(count == 0) {
            return; // x和y都找到了就终止递归
        }
        // 左孩子节点
        if(node.left != null) {
            if(node.left.val == x || node.left.val == y) {
                list.add(node.val); // 父节点的值
                list.add(deep+1); // 当前节点的深度
                --count;
                return; // 提前终止该分支的递归
            }
            dfs(node.left, x, y, deep+1);
        }
        // 右孩子节点
        if(node.right != null) {
            if(node.right.val == x || node.right.val == y) {
                list.add(node.val);
                list.add(deep+1);
                --count;
                return;
            }
            dfs(node.right, x, y, deep+1);
        }
    }
}

注意点

• 此题不用将树全部遍历完，有许多情况可以提前结束当前分支的递归，甚至是结束全部递归