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、题目描述:
-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的栈顶元素即为最小值
牛客题霸 文章被收录于专栏
本专栏主要是牛客题霸习题的讲解,有详细的考点分类,大家可以可以看看呦!!!