从二叉搜索树到更大和树

/*
题目思路：
- 使用Stream API的方式来实现相同的逻辑。
- 由于Stream API主要用于操作集合类，通常不会直接处理树结构，因此，这里提供了一种模拟Stream API风格的代码。
*/

import java.util.ArrayList;
import java.util.List;

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

public class Solution {
    private int sum = 0;
    
    public TreeNode convertBST(TreeNode root) {
        List<TreeNode> list = new ArrayList<>();
        traverseAndCollect(root, list);
        list.stream().sorted((a, b) -> b.val - a.val).forEach(node -> {
            sum += node.val;
            node.val = sum;
        });
        return root;
    }
    
    private void traverseAndCollect(TreeNode node, List<TreeNode> list) {
        if (node == null) return;
        traverseAndCollect(node.left, list);
        list.add(node);
        traverseAndCollect(node.right, list);
    }
}