题解 | #设计getMin功能的栈#

设计getMin功能的栈

http://www.nowcoder.com/practice/05e57ce2cd8e4a1eae8c3b0a7e9886be

设计getMin功能的栈

  • 创建两个栈 一个栈stack_data存所有数据,一个栈stack_min【栈顶最小】存实时最小值
  • 进栈push: stack_min将每个数据与栈顶数据比较(空栈直接进来) 小于栈顶则进来
  • 出栈pop: stack_data出栈数据和 stack_min栈顶比较(只有可能>=) =直接 stack_min直接出栈
  • getMin 直接出实时最小 stack_min 栈顶元素
import java.util.*;
import java.io.*;
public class Main{
    static Stack<Integer> stack_data = new Stack<>();
    static Stack<Integer> stack_min = new Stack<>();
    
    //栈空 或者 stack_min栈顶比较进栈数据小者存进来
    public static void _push(int num){
        if(stack_min.isEmpty()){
            stack_min.push(num);
        }else if(num <= stack_min.peek()){
            stack_min.push(num);
        }
        stack_data.push(num);
    }
    //出栈 如果出栈 stack_data 的数据和  stack_min 相同 则stack_min出栈
    public static void _pop(){
        if(stack_data.isEmpty()){
            System.out.println("栈空");
            return;
        }
        int current = stack_data.pop();
        if(current == stack_min.peek()){
            stack_min.pop();
        }
    }
    
    public static void main(String args[]) throws IOException{
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.valueOf(in.readLine());
        for(int i = 0; i < n; i++){
            String ins[] = in.readLine().split(" ");
            String str = ins[0];
            if("push".equals(str)){
                 int num = Integer.parseInt(ins[1]);
                _push(num);
            }else if("pop".equals(str)){
                _pop();
            }else{
                System.out.println(stack_min.peek());
            }
        }
        in.close();
    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务