232. 用栈实现队列

232. 用栈实现队列


有个基本一样的题目 剑指 Offer 09. 用两个栈实现队列

题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作(push、pop、peek、empty):

实现 MyQueue 类:

  • void push(int x) 将元素 x 推到队列的末尾
  • int pop() 从队列的开头移除并返回元素
  • int peek() 返回队列开头的元素
  • boolean empty() 如果队列为空,返回 true ;否则,返回 false

说明:

  • 你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
  • 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
    进阶:

进阶:

你能否实现每个操作均摊时间复杂度为 O(1) 的队列?换句话说,执行 n 个操作的总时间复杂度为 O(n) ,即使其中一个操作可能花费较长时间。

示例:

输入:
[“MyQueue”, “push”, “push”, “peek”, “pop”,“empty”]
[[], [1], [2],[], [], []]
输出:
[null, null, null, 1, 1, false]

解释:

MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false

提示:

1 <= x <= 9
最多调用 100 次 push、pop、peek 和 empty
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)

方法一:双栈

思路

将一个栈当作输入栈,用于压入 push 传入的数据;另一个栈当作输出栈,用于 pop 和 peek 操作。

每次 pop 或 peek 时,若输出栈为空则将输入栈的全部数据依次弹出并压入输出栈,这样输出栈从栈顶往栈底的顺序就是队列从队首往队尾的顺序。

C++:

class MyQueue
{
private:
	stack<int> inStack, outStack;
	void int2Out()
	{
		while (!inStack.empty())
		{
			outStack.push(inStack.top());
			inStack.pop();
		}
	}
public:
	/** Initialize your data structure here. */
	MyQueue()
	{
		while (!inStack.empty())//初始化堆栈,清空处理
		{
			inStack.pop();
		}
		while (!outStack.empty())
		{
			outStack.pop();
		}
	}

	/** Push element x to the back of queue. */
	void push(int x)
	{
		inStack.push(x);
	}

	/** Removes the element from in front of queue and returns that element. */
	int pop()
	{
		if (outStack.empty())
		{
			int2Out();
		}
		if (outStack.empty())
		{
			return -1;
		}
		int t = outStack.top();
		outStack.pop();
		return t;
	}

	/** Get the front element. */
	int peek()
	{
		if (outStack.empty())
		{
			int2Out();
		}
		return outStack.top();
	}

	/** Returns whether the queue is empty. */
	bool empty()
	{
		return inStack.empty() && outStack.empty();
	}
};

C语言:

typedef struct  
{
	int* stk;
	int stkSize;
	int stkCapacity;
}Stack;

Stack* stackCreate(int capacity)
{
	Stack* ret = malloc(sizeof(Stack));
	ret->stk = malloc(sizeof(int)*capacity);
	ret->stkSize = 0;
	ret->stkCapacity = capacity;
	return ret;
}

void stackPush(Stack* obj, int x)
{
	obj->stk[obj->stkSize++] = x;
}
void stackPop(Stack* obj)
{
	obj->stkSize--;//这里的写法有点出乎意料,直接把stkSize--就行了,stkSize里的数据在调用free时再被清除  
}
int stackTop(Stack* obj)
{
	return obj->stk[obj->stkSize - 1];
}
bool stackEmpty(Stack* obj)
{
	return obj->stkSize == 0;
}
void stackFree(Stack* obj)
{
	free(obj->stk);
	free(obj);
}
typedef struct 
{
	Stack* inStack;
	Stack* outStack;
}MyQueue;
MyQueue* myQueueCreate()
{
	MyQueue* ret = malloc(sizeof(MyQueue));
	ret->inStack = stackCreate(100);
	ret->outStack = stackCreate(100);
	return ret;
}
void in2Out(MyQueue* obj)
{
	while (!stackEmpty(obj->inStack))
	{
		stackPush(obj->outStack, stackTop(obj->inStack));
		stackPop(obj->inStack);
	}
}
void myQueuePush(MyQueue* obj,int x)
{
	stackPush(obj->inStack, x);
}
int myQueuePop(MyQueue* obj)
{
	if (stackEmpty(obj->outStack))
	{
		in2Out(obj);
	}
	int temp = stackTop(obj->outStack);
	stackPop(obj->outStack);
	return temp;
}
int myQueuePeek(MyQueue* obj)
{
    if(stackEmpty(obj->outStack))
    {
        in2Out(obj);
    }
	return stackTop(obj->outStack);
}
bool myQueueEmpty(MyQueue* obj)
{
	return stackEmpty(obj->inStack) && stackEmpty(obj->outStack);
}
void myQueueFree(MyQueue* obj)
{
	stackFree(obj->inStack);
	stackFree(obj->outStack);
    free(obj);
}

复杂度分析

时间复杂度:push 和 empty 为 O(1),pop 和 peek 为均摊 O(1)。对于每个元素,至多入栈和出栈各两次,故均摊复杂度为 O(1)。

空间复杂度:O(n)。其中 n 是操作总数。对于有 n 次 push 操作的情况,队列中会有 n 个元素,故空间复杂度为 O(n)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

龙叙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值