232. 用栈实现队列

题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（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（双端队列）来模拟一个栈，只要是标准的栈操作即可。

解题思路

#![allow(unused)]


fn main() {
struct MyQueue {
    is: Vec<i32>,
    os: Vec<i32>,
}


/**
 * `&self` means the method takes an immutable reference.
 * If you need a mutable reference, change it to `&mut self` instead.
 */
impl MyQueue {

    fn new() -> Self {
        Self { is: vec![], os: vec![] }
    }

    fn o2i(&mut self) {
        while let Some(i) = self.os.pop() {
            self.is.push(i);
        }
    }

    fn i2o(&mut self) {
        while let Some(i) = self.is.pop() {
            self.os.push(i);
        }
    }
    
    fn push(&mut self, x: i32) {
        self.o2i();
        self.is.push(x);
    }
    
    fn pop(&mut self) -> i32 {
        self.i2o();
        self.os.pop().unwrap()
    }
    
    fn peek(&mut self) -> i32 {
        self.i2o();
        self.os.last().copied().unwrap()
    }
    
    fn empty(&self) -> bool {
        self.is.is_empty() && self.os.is_empty()
    }
}
}

学习感想