K 次操作后最大化顶端元素

/*
 * 题目思路：
 * 1. 使用Java Stream API将数组转化为栈。
 * 2. 根据k的值决定操作类型，并维护一个暂存删除元素的列表。
 * 3. 如果k的值大于等于nums的长度，返回-1。
 * 4. 否则通过比较暂存的删除元素，找到最大值。
 */

import java.util.Stack;
import java.util.ArrayList;
import java.util.Arrays;

public class MaxStackTopStream {
    public int getMaxStackTopStream(int[] nums, int k) {
        if (k >= nums.length) return -1;
        Stack<Integer> stack = new Stack<>();
        ArrayList<Integer> removedElements = new ArrayList<>();
        Arrays.stream(nums).forEach(stack::push);
        for (int i = 0; i < k; i++) {
            if (!stack.isEmpty()) {
                removedElements.add(stack.pop());
            } else if (!removedElements.isEmpty()) {
                stack.push(removedElements.remove(removedElements.size() - 1));
            }
        }
        return stack.isEmpty() ? -1 : stack.peek();
    }

    public static void main(String[] args) {
        MaxStackTopStream mst = new MaxStackTopStream();
        System.out.println(mst.getMaxStackTopStream(new int[]{5,2,2,4,0,6}, 4)); // 输出: 5
        System.out.println(mst.getMaxStackTopStream(new int[]{2}, 1)); // 输出: -1
    }
}