【代码随想录算法训练营14期】Day10- 第五章 栈与队列part01

Day10- 第五章 栈与队列part01


文章目录

  • Day10- 第五章 栈与队列part01
  • 232. 用栈实现队列
  • 解题思路
    • 1.代码
  • 225. 用队列实现栈
  • 解题思路
    • 2.代码
  • 总结


232. 用栈实现队列

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

实现 MyQueue 类:

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


解题思路

栈实现队列,使用一个输入栈来存储,入队列即把元素放入输入栈;从队列移除元素,即判断输出栈是否为空,不为空直接输出栈顶即为队列开头元素;如果输出栈为空,则读取所有输入栈的元素放入输出栈。

1.代码

class MyQueue {
public:
    stack<int> In;//输入栈
    stack<int> Out;//输出栈
    MyQueue() {

    }
    void push(int x) {
        In.push(x);
    }
    
    int pop() {
        if(Out.empty()){
            while(!In.empty()){
                Out.push(In.top());
                In.pop();
            }
        }
        int x = Out.top();
        Out.pop();
        return x;
    }
    
    int peek() {
          if(Out.empty()){
            while(!In.empty()){
                Out.push(In.top());
                In.pop();
            }
        }
        return Out.top();
    }
    
    bool empty() {
        if(In.empty() && Out.empty())
            return true;
        else
            return false;
    }
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */

225. 用队列实现栈

请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。

实现 MyStack 类:

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。


解题思路

队列实现栈,唯一不同的是每放入一个元素x,就要把队列的前面的元素全部读出,再入队列,这样刚入队列的元素就在队首

2.代码

class MyStack {
public:
    queue<int> q;
    MyStack() {

    }
    void push(int x) {
        q.push(x);
        int n = q.size();
        for(int i = 0; i < n - 1; i++){
            int tmp = q.front();
            q.pop();
            q.push(tmp);
        }
    }
    
    int pop() {
        int x = q.front();
        q.pop();
        return x;
    }
    
    int top() {
        return q.front();
    }
    
    bool empty() {
        return q.empty();
    }
};

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack* obj = new MyStack();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->top();
 * bool param_4 = obj->empty();
 */

总结

  • 栈stack s1;栈顶top(),pop();队列queue q1;front()

你可能感兴趣的:(代码随想录算法训练营,c++,java,数据结构)