二叉搜索树中的中序后继 II

/*
 * 思路：
 * 1. 使用递归函数查找二叉树中的后继节点。
 * 2. 在函数中使用Stream API来查找右子树中的最左节点。
 * 3. 在没有右子树的情况下，使用一个辅助函数递归查找可能的后继。
 */
import java.util.Optional;
import java.util.function.Function;
 
public class Solution {
    public Node inorderSuccessor(Node node) {
        if (node == null) return null;
        // Case 1: Node has a right subtree
        if (node.right != null) {
            return Optional.of(node.right)
                           .map(this::findMin)
                           .orElse(null);
        }
        // Case 2: Node doesn't have a right subtree
        return findSuccessor(null, node);
    }
 
    private Node findMin(Node node) {
        return Optional.ofNullable(node)
                       .map(n -> {
                           while (n.left != null) n = n.left;
                           return n;
                       })
                       .orElse(null);
    }
 
    private Node findSuccessor(Node successor, Node node) {
        if (node == null) return successor;
        if (node.val < node.val) {
            successor = node;
            return findSuccessor(successor, node.left);
        } else if (node.val > node.val) {
            return findSuccessor(successor, node.right);
        } else {
            return successor;
        }
    }
}