c# - Stack 中元素的弹出范围

Question

大家好，我需要一些关于 Stack.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}