Stack中元素的弹出范围

Question

嗨，伙计们，我需要一个关于堆栈的小帮助。Pop（）函数。我知道堆栈可以一个接一个地弹出元素，但是我需要不止一个元素才能弹出。例如，我在堆栈中有5个元素（4,3,2,1,0），现在我想弹出前3个或2个元素，直到堆栈索引达到1或2。现在我有“for”循环，它不能正常工作：

for(var i = stack.Count - 1; i >= 0; i--)
{
    stack.Pop();
}

有人能帮我吗，让他弹出一定范围的元素？谢谢！

Answer 1

您还可以创建一个扩展方法：

public static class Extensions
{
    public static List<T> PopRange<T>(this Stack<T> stack, int amount)
    {
        var result = new List<T>(amount);
        while (amount-- > 0 && stack.Count > 0)
        {
            result.Add(stack.Pop());
        }
        return result;
    }
}

并在您想要的地方使用它：

var stack = new Stack<int>(new[] { 1, 2, 3, 4, 5 });

var result = stack.PopRange(3);

// result: { 5, 4, 3 }
//  stack: { 2, 1}

Answer 2

您可以使用一个简单的while循环来执行以下操作：

var stack = new Stack<int>(new[]{ 4, 3, 2, 1, 0 });

var numberToPop = 3;
while(numberToPop > 0 && stack.Count > 0)
{
    numberToPop--;
    stack.Pop();
}

Answer 3

如果你想弹出直到堆栈达到一定大小，只需使用

while(stack.Count > desiredCount)
    stack.Pop();

如果你想弹出一定数量的物品，那就使用

for(int i=0; i < numberOfItemsToPop && stack.Count > 0; i++)
    stack.Pop();