二叉树的堂兄弟节点

/*
 * 思路：
 * 1. 通过DFS遍历树并使用stream API来记录每个节点的深度和父节点。
 * 2. 使用递归函数来更新深度和父节点。
 * 3. 检查两个节点的深度是否相同且父节点不同。
 */
import java.util.*;
import java.util.stream.*;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    public boolean isCousins(TreeNode root, int x, int y) {
        Map<Integer, Integer> depth = new HashMap<>();
        Map<Integer, TreeNode> parent = new HashMap<>();
        dfs(root, null, depth, parent, 0);
        return depth.get(x) == depth.get(y) && parent.get(x) != parent.get(y);
    }

    private void dfs(TreeNode node, TreeNode par, Map<Integer, Integer> depth, Map<Integer, TreeNode> parent, int d) {
        if (node == null) return;
        depth.put(node.val, d);
        parent.put(node.val, par);
        dfs(node.left, node, depth, parent, d + 1);
        dfs(node.right, node, depth, parent, d + 1);
    }
}