代码随想录第10天栈与队列part01|232.用栈实现队列、225. 用队列实现栈
理论基础
队列:先进先出
栈:先进后出
在C++中,栈和队列是STL(C++标准库)里面的两个数据结构,三个最为普遍的STL版本:
- HP STL 其他版本的C++ STL,一般是以HP STL为蓝本实现出来的,HP STL是C++ STL的第一个实现版本,而且开放源代码。
- P.J.Plauger STL 由P.J.Plauger参照HP STL实现出来的,被Visual C++编译器所采用,不是开源的。
- SGI STL 由Silicon Graphics Computer Systems公司参照HP STL实现,被Linux的C++编译器GCC所采用,SGI STL是开源软件,源码可读性甚高。
接下来介绍的栈和队列也是SGI STL里面的数据结构, 知道了使用版本,才知道对应的底层实现。
栈
栈提供push 和 pop 等等接口,所有元素先进后出,所以栈不提供走访功能,也不提供迭代器(iterator)。 不像是set 或者map 提供迭代器iterator来遍历所有元素。
232.用栈实现队列
思路
:
用两个栈(输入栈、输出栈)实现一个队列的行为
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来,再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
peek()的实现注意可以复用pop()
伪代码
:
stack in stack out
void push(int x){
stackIn.push(x);
}
int pop(
if(stackOut.empty()){ \\ 出栈为空
while(! stackIn.empty()){ \\ 入栈不为空
stackOut.push(stackIn.top());
stackIn.pop();
}
result = stackOut.top();
stackOut.pop();
return result;
}
python代码
:
Python List pop()方法:
pop() 函数用于移除列表中的一个元素(默认最后一个元素),并且返回该元素的值。
刚开始不理解def peek(self) 中的ans = self.pop()
该类的测试代码:
obj = MyQueue()
obj.push(2)
obj.push(3)
obj.peek()
调用obj.peek()时,相当于先调用了obj.pop(),
ans为出栈的值
class MyQueue:
def __init__(self):
self.stack_in = []
self.stack_out = []
def push(self, x: int) -> None:
self.stack_in.append(x)
def pop(self) -> int:
if self.empty():
return None
if self.stack_out:
return self.stack_out.pop()
else:
for i in range(len(self.stack_in)):
self.stack_out.append(self.stack_in.pop())
return self.stack_out.pop()
def peek(self) -> int:
ans = self.pop()
self.stack_out.append(ans)
return ans
def empty(self) -> bool:
return not (self.stack_in or self.stack_out)
# Your MyQueue object will be instantiated and called as such:
# obj = MyQueue()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.peek()
# param_4 = obj.empty()
225. 用队列实现栈
思路
:
用一个队列即可实现栈的进出元素,将前面的元素取出来后再加入队列中,弹出队列元素数量-1个元素。
Python普通的Queue或SimpleQueue没有类似于peek的功能,也无法用索引访问,在实现top的时候较为困难。用list可以,但是在使用pop(0)的时候时间复杂度为O(n)。
因此这里使用双向队列,我们保证只执行popleft()和append(),因为deque可以用索引访问,可以实现和peek相似的功能
python代码
:
class MyStack:
def __init__(self):
self.que = deque()
def push(self, x: int) -> None:
self.que.append(x)
def pop(self) -> int:
if self.empty():
return None
for i in range(len(self.que)-1):
self.que.append(self.que.popleft())
return self.que.popleft()
def top(self) -> int:
if self.empty():
return None
for i in range(len(self.que)-1):
self.que.append(self.que.popleft())
temp = self.que.popleft()
self.que.append(temp)
return temp
def empty(self) -> bool:
return not self.que
# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()