在堆栈中的某个位置插入元素的一种方法。(java)

在Java中，堆栈（Stack）是一种遵循后进先出（LIFO, Last In First Out）原则的数据结构。它允许我们在其顶部添加和移除元素。要在堆栈中的某个位置插入元素，通常的做法是先将所有位于目标位置之上的元素弹出，然后在目标位置插入新元素，最后将之前弹出的元素重新压入堆栈。

然而，Java的Stack类并没有直接提供在中间插入元素的方法，因为它主要是为了实现LIFO行为而设计的。如果你需要在堆栈中间插入元素，你可以考虑使用ArrayDeque类，它实现了Deque接口，提供了更灵活的双端队列操作。

以下是一个使用ArrayDeque在中间插入元素的示例代码：

import java.util.ArrayDeque;
import java.util.Deque;

public class StackInsertExample {
    public static void main(String[] args) {
        Deque<Integer> stack = new ArrayDeque<>();
        
        // 向堆栈中添加一些元素
        stack.push(1);
        stack.push(2);
        stack.push(3);
        stack.push(4);
        
        System.out.println("原始堆栈: " + stack);
        
        int insertElement = 99;
        int position = 2; // 插入位置，从0开始计数
        
        // 在指定位置插入元素
        insertAtPosition(stack, insertElement, position);
        
        System.out.println("插入元素后的堆栈: " + stack);
    }
    
    public static void insertAtPosition(Deque<Integer> stack, int element, int position) {
        // 创建一个临时堆栈来保存弹出的元素
        Deque<Integer> tempStack = new ArrayDeque<>();
        
        // 弹出目标位置之上的所有元素
        for (int i = 0; i < position; i++) {
            tempStack.push(stack.pop());
        }
        
        // 在目标位置插入新元素
        stack.push(element);
        
        // 将之前弹出的元素重新压入堆栈
        while (!tempStack.isEmpty()) {
            stack.push(tempStack.pop());
        }
    }
}

在这个示例中，我们首先创建了一个ArrayDeque对象作为堆栈，并向其中添加了一些元素。然后，我们定义了一个insertAtPosition方法，该方法接受堆栈、要插入的元素和插入位置作为参数。在方法内部，我们使用一个临时堆栈来保存弹出的元素，然后在目标位置插入新元素，最后将之前弹出的元素重新压入堆栈。

需要注意的是，这种方法的时间复杂度是O(n)，其中n是堆栈中的元素数量，因为我们需要弹出和重新压入元素。如果你需要频繁地在堆栈中间插入元素，可能需要考虑使用其他数据结构，如链表。

参考链接：