找到二叉树中最近的右侧节点

/*
题目思路：
1. 使用层序遍历和队列来遍历二叉树。
2. 使用Java Stream API从队列中过滤并找到目标节点。
3. 一旦找到目标节点，返回其右侧的节点，如果没有右侧节点，返回null。
*/

import java.util.*;
import java.util.stream.*;

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

class Solution {
    public TreeNode findNearestRightNode(TreeNode root, TreeNode u) {
        if (root == null) return null;

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);

        while (!queue.isEmpty()) {
            List<TreeNode> levelNodes = new ArrayList<>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode currentNode = queue.poll();
                levelNodes.add(currentNode);

                if (currentNode.left != null) queue.offer(currentNode.left);
                if (currentNode.right != null) queue.offer(currentNode.right);
            }

            Optional<TreeNode> node = levelNodes.stream().filter(n -> n == u).findFirst();
            if (node.isPresent()) {
                int index = levelNodes.indexOf(node.get());
                return index == levelNodes.size() - 1 ? null : levelNodes.get(index + 1);
            }
        }

        return null; // 如果遍历完都没有找到，返回null
    }
}