C++ 用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/aaa958099161/article/details/89974832

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

提交代码:

class Solution
{
public:
	//stack1 作为中间转运容器
	void push(int node) {
		while (!stack2.empty())
		{
			stack1.push(stack2.top());
			stack2.pop(); //删除
		}
		stack2.push(node);
		while (!stack1.empty())
		{
			stack2.push(stack1.top());
			stack1.pop();
		}
	}

	int pop() {
		if (stack2.empty())
		{
			return 0;
		}
		int out = stack2.top();	//不能直接返回
		stack2.pop();   //从stack中删除
		return out;
	}

private:
	stack<int> stack1;
	stack<int> stack2;
};

测试代码:

int main()
{
	Solution *s = new Solution();
	for (int i=0;i<10;i++)
	{
		s->push(i);
		if (i==5)
		{
			s->pop();
		}
	}

	system("pause");
	return 0;
}

猜你喜欢

转载自blog.csdn.net/aaa958099161/article/details/89974832