剑指Offer——用两个栈实现队列

最新推荐文章于 2022-04-02 10:48:00 发布

原创最新推荐文章于 2022-04-02 10:48:00 发布 · 179 阅读

0 ·

CC 4.0 BY-SA版权

本文为叶琼州原创文章，转载请加链接https://yeqiongzhou.com

文章标签：

#题解 #剑指offer

剑指offer题解专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种使用两个栈实现队列的方法，以及利用两个队列实现栈的技术。通过代码示例详细解释了如何进行Push和Pop操作，展示了在不同数据结构间转换的灵活性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

题解

#include <iostream>
#include <stack>
#include <queue>

using namespace std;

stack<int> stack1, stack2;
queue<int> queue1, queue2;


void push1(int node)
{
	stack1.push(node);
}

int pop1()
{
	int node;
	if (stack2.empty())
	{
		while (!stack1.empty())
		{
			node = stack1.top();
			stack2.push(node);
			stack1.pop();
		}
	}
	node = stack2.top();
	stack2.pop();
	return node;
}

void push2(int node)
{
	queue1.push(node);
}

int pop2()
{
	int node;
	while (queue1.size() > 1)
	{
		int temp = queue1.front();
		queue1.pop();
		queue2.push(temp);
	}
	node = queue1.front();
	queue1.pop();
	while (queue2.size())
	{
		int temp = queue2.front();
		queue2.pop();
		queue1.push(temp);
	}
	return node;
}

int main()
{
	ios::sync_with_stdio(false);
	int n, temp;
	cin >> n;
	cout << "用两个栈实现一个队列：" << endl;
	for (int i = 0; i < n; i++)
	{
		cin >> temp;
		push1(temp);
	}
	while (!stack1.empty() || !stack2.empty())
	{
		cout << pop1() << " ";
	}
	cout << endl << "用两个队列实现一个栈：" << endl;
	for (int i = 0; i < n; i++)
	{
		cin >> temp;
		push2(temp);
	}
	while (!queue1.empty())
	{
		cout << pop2() << " ";
	}
	cout << endl;
	return 0;
}