NC90 设计getMin功能的栈

设计getMin功能的栈

https://www.nowcoder.com/practice/c623426af02d4c189f92f2a99647bd34?tpId=188&tqId=37556&rp=1&ru=%2Fta%2Fjob-code-high-week&qru=%2Fta%2Fjob-code-high-week%2Fquestion-ranking&tab=answerKey

- 1、题目描述:
图片说明

- 2、题目链接:
https://www.nowcoder.com/practice/c623426af02d4c189f92f2a99647bd34?tpId=188&tqId=37556&rp=1&ru=%2Fta%2Fjob-code-high-week&qru=%2Fta%2Fjob-code-high-week%2Fquestion-ranking&tab=answerKey

-3、 设计思想:
图片说明
详细操作流程看下图:
图片说明

-4、视频讲解链接B站视频讲解

-5、代码:
c++版本:

 class Solution {
public:
    /**
     * return a array which include all ans for op3
     * @param op int整型vector<vector<>> operator
     * @return int整型vector
     */
     stack<int> s,min_s;//s用来存数据,min_s用来获取最小值
    vector<int> getMinStack(vector<vector<int> >& op) {
        // write code here
        vector<int> res;//结果集
        for (int i = 0; i < op.size(); i++) {
            if(op[i][0] == 1)//为1就入栈
                Push(op[i][1]);
            else if (op[i][0] == 2)//为2就出栈
                Pop();
            else //为3就返回栈中最小元素
                res.push_back(getMin());
        }
        return res;

    }
    void Push(int x){
        s.push(x);
        if(min_s.empty() || min_s.top() >= x) min_s.push(x);//如果min_s为空,或者栈顶元素大于x就进入min_s

    }
    void Pop() {
        if(!s.empty()){
            if(s.top() == min_s.top()) min_s.pop();//如果min_s栈顶元素和栈s中要出栈的元素相等,那么也需要出栈
            s.pop();

        }
    }
    int getMin(){
        return min_s.top();//栈min_s的栈顶元素即为最小值
    }
};

Java版本:

import java.util.*;


public class Solution {
    /**
     * return a array which include all ans for op3
     * @param op int整型二维数组 operator
     * @return int整型一维数组
     */
    public static Stack<Integer>s = new Stack<Integer>();//s用来存数据
    public static Stack<Integer>min_s = new Stack<Integer>();//min_s用来获取最小值
    public int[] getMinStack (int[][] op) {
        // write code here
        ArrayList<Integer> res = new ArrayList<Integer>();//结果集
        for (int i = 0;i<op.length;i++){
            if (op[i][0] == 1){ //为1就入栈
                Push(op[i][1]);
            }else if (op[i][0] == 2){//为2就出栈
                Pop();
            }else//为3就返回栈中最小元素
                res.add(getMin());
        }
        //因为返回类型的缘故我们需要把res中的元素丢在arr数组里面
        int [] arr = new int[res.size()];
        for (int i = 0;i< res.size();i++){
            arr[i] = res.get(i);
        }
        return arr;

    }
    public void Push(int x){
        s.push(x);
        if(min_s.empty() || min_s.peek()>=x)min_s.push(x); //如果min_s为空,或者栈顶元素大于x就进入min_s
    }
    public void Pop() {
        if(!s.empty()){
            if(s.peek().equals(min_s.peek())) min_s.pop();//如果min_s栈顶元素和栈s中要出栈的元素相等,那么也需要出栈
            s.pop();
        }   

    }
    int getMin(){
        return min_s.peek();//栈min_s的栈顶元素即为最小值
    }

}

Python版本:

#
# return a array which include all ans for op3
# @param op int整型二维数组 operator
# @return int整型一维数组
#
class Solution:
    s = []
    min_s = []
    def getMinStack(self , op ):
        # write code here
        res = []
        for i in range(len(op)):
            if op[i][0] == 1:#为1就入栈
                self.Push(op[i][1])
            elif op[i][0] == 2:#为2就出栈
                self.Pop()
            else:#为3就返回栈中最小元素
                res.append(self.getMin())
        return res

    def Push(self,x):
        self.s.append(x)
        if not self.min_s or self.min_s[-1] >= x: self.min_s.append(x) #如果min_s为空,或者栈顶元素大于x就进入min_s

    def Pop(self):
        if self.s:
            if  self.s[-1]== self.min_s[-1] : self.min_s.pop() #如果min_s栈顶元素和栈s中要出栈的元素相等,那么也需要出栈
            self.s.pop()
    def getMin(self):
        return self.min_s[-1] #栈min_s的栈顶元素即为最小值
牛客题霸 文章被收录于专栏

本专栏主要是牛客题霸习题的讲解,有详细的考点分类,大家可以可以看看呦!!!

全部评论
请问为什么为1入栈,为2出栈,为3返回栈顶元素呀?
点赞 回复 分享
发布于 2021-03-07 19:30

相关推荐

安静的垂耳兔在泡澡:ks已经第八次投递了,它起码挂了还让你再投,不错了
点赞 评论 收藏
分享
评论
8
1
分享
牛客网
牛客企业服务