文章

225-implement-stack-using-queues

225-implement-stack-using-queues

Implement a last-in-first-out (LIFO) stack using only two queues. The implemented stack should support all the functions of a normal stack (push, top, pop, and empty).

Implement the MyStack class:

void push(int x) Pushes element x to the top of the stack. int pop() Removes the element on the top of the stack and returns it. int top() Returns the element on the top of the stack. boolean empty() Returns true if the stack is empty, false otherwise. Notes:

You must use only standard operations of a queue, which means that only push to back, peek/pop from front, size and is empty operations are valid. Depending on your language, the queue may not be supported natively. You may simulate a queue using a list or deque (double-ended queue) as long as you use only a queue’s standard operations.

Example 1:

Input [“MyStack”, “push”, “push”, “top”, “pop”, “empty”] [[], [1], [2], [], [], []] Output [null, null, null, 2, 2, false]

Explanation MyStack myStack = new MyStack(); myStack.push(1); myStack.push(2); myStack.top(); // return 2 myStack.pop(); // return 2 myStack.empty(); // return False

My solution:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from collections import deque
class MyStack(object):

    def __init__(self):
        self.mainQuene = deque()
        self.subQuene = deque()

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.mainQuene.append(x)
    def pop(self):  
        """
        :rtype: int
        """
        while len(self.mainQuene) > 1:
            self.subQuene.append(self.mainQuene.popleft())
        res = self.mainQuene.popleft()
        while len(self.subQuene) > 0:
            self.mainQuene.append(self.subQuene.popleft())
        return res
        

    def top(self):
        """
        :rtype: int
        """
        while len(self.mainQuene) > 1:
            self.subQuene.append(self.mainQuene.popleft())
        res = self.mainQuene.popleft()
        self.subQuene.append(res)
        while len(self.subQuene) > 0:
            self.mainQuene.append(self.subQuene.popleft())
        return res
        
        

    def empty(self):
        """
        :rtype: bool
        """
        return True if len(self.mainQuene) == 0 else False


# 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()
本文由作者按照 CC BY 4.0 进行授权