移除堆栈中的特定元素

Question

我试图删除堆栈中的特定元素，但遇到了一些问题。我的想法是将元素弹出到一个临时堆栈中，弹出我正在寻找的索引，然后将临时堆栈中的元素弹出到主堆栈中。我很难想象如何让临时堆栈回到顶部。任何帮助都将不胜感激。

public E remove(int index) {
    Stack<E> tmpStack = new Stack<E>();
    if (size() == 0) {
        return null;
    } else {
        for (int i = 0; i < index; i++) {
            tmpStack.push(this.pop());
        }
        return tmpStack.pop();
    }
    while (!tmpStack.isEmpty())
        this.push(tmpStack.pop());
}

想法？干杯！

Answer 1

访问堆栈中特定索引处的元素不利于拥有堆栈。其他考虑因素包括索引是否引用堆栈的最顶部。正如其他人提到的，问题是您在将元素放回原始堆栈之前返回。您还应该考虑索引大于堆栈当前大小的情况，因为您当前的实现将导致错误。（JavaStack将抛出一个空StackExctive）

Answer 2

问题是，在最后一次还原操作之前有一个返回，所以方法返回时包含已删除的元素，而不在其后面调用代码。

通常会出现无法访问的代码错误，但在您的情况下，这不是真的，因为您没有将最后一个while括在else分支中，因此如果堆栈为空，则执行while（即使在空堆栈上），java编译器无法检测到这一点。

你应该做类似的事情：

if (isEmpty())
  return null;
else
{
  for (int i = 0; i < index; i++)
    tmpStack.push(this.pop());

  E removedElement = tmpStack.pop();

  while (!tmpStack.isEmpty())
    this.push(tmpStack.pop());

  return removedElement;
}