AcWing 828. 模拟栈

最新推荐文章于 2024-05-30 13:16:30 发布

原创最新推荐文章于 2024-05-30 13:16:30 发布 · 218 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #栈 #数据结构

AcWing算法基础专栏收录该内容

78 篇文章

订阅专栏

本文介绍了一个使用Java实现的栈数据结构，支持push、pop、empty和query四种操作。通过对10万次操作的处理，展示了如何在程序中有效地进行栈的管理和查询。输入包括操作次数和具体操作命令，输出则是对empty和query操作的响应。

实现一个栈，栈初始为空，支持四种操作：

(1) “push x” – 向栈顶插入一个数x；

(2) “pop” – 从栈顶弹出一个数；

(3) “empty” – 判断栈是否为空；

(4) “query” – 查询栈顶元素。

现在要对栈进行M个操作，其中的每个操作3和操作4都要输出相应的结果。

输入格式

第一行包含整数M，表示操作次数。

接下来M行，每行包含一个操作命令，操作命令为”push x”，”pop”，”empty”，”query”中的一种。

输出格式

对于每个”empty”和”query”操作都要输出一个查询结果，每个结果占一行。

其中，”empty”操作的查询结果为“YES”或“NO”，”query”操作的查询结果为一个整数，表示栈顶元素的值。

数据范围

1≤M≤1000001≤M≤100000,
1≤x≤1091≤x≤109
所有操作保证合法。

输入样例：

10
push 5
query
push 6
pop
query
pop
empty
push 4
query
empty

输出样例：

5
5
YES
4
NO

import java.io.*;
import java.lang.Integer;

class Main{
    static int N = 100010;
    static int[] st = new int[N];
    static int tt = 0;
    
    static void push(int x){
        st[++tt] = x;
    }
    static void pop(){
        if(!empty())
            tt--;
    }
    static boolean empty(){
        if(tt > 0)return false;
        return true;
    }
    static int query(){
        if(!empty())
            return st[tt];
        return -1;
    }
    
    
    public static void main(String[] args)throws Exception{
        BufferedReader buf = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter pw = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.valueOf(buf.readLine());
        for(int i = 0; i < n; ++i){
            String[] params = buf.readLine().split(" ");
            if("push".equals(params[0])){
                int x = Integer.valueOf(params[1]);
                push(x);
            }else if("pop".equals(params[0])){
                pop();
            }else if("empty".equals(params[0])){
                boolean res = empty();
                if(res)pw.write("YES\n");
                else pw.write("NO\n");
            }else{
                int res = query();
                if(res > 0)pw.write(res + "\n");
            }
        }
        pw.flush();
        buf.close();
        pw.close();
    }
}