装饰树

/*
 * 题目思路：
 * 使用Java Stream API，我们可以创建一种更为函数式的方式来处理这个问题。
 * 通过Stream，我们可以使用递归来处理树的每个节点。
 * 对于每一个节点，如果它有左子节点，则在它和左子节点之间插入一个值为-1的节点；
 * 如果它有右子节点，则在它和右子节点之间插入一个值为-1的节点。
 */

import java.util.function.Function;

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

public class Solution {
    public TreeNode addDecorations(TreeNode root) {
        if (root == null) return null;
        Function<TreeNode, TreeNode> decorate = (TreeNode node) -> {
            if (node == null) return null;
            if (node.left != null) {
                TreeNode left = node.left;
                node.left = new TreeNode(-1);
                node.left.left = decorate.apply(left);
            }
            if (node.right != null) {
                TreeNode right = node.right;
                node.right = new TreeNode(-1);
                node.right.right = decorate.apply(right);
            }
            return node;
        };
        return decorate.apply(root);
    }
}