[백준 알고리즘] 10828번 스택 (Python, Java)

 

<Python>

import sys
N = int(input())
stack = []

for _ in range(N):
    line = sys.stdin.readline().split()

    if line[0] == "push":
        stack.append(int(line[1]))

    elif line[0] == "pop":
        if len(stack)>0:
            print(stack[-1])
            stack.pop()
        else:
            print(-1)

    elif line[0] == "size":
        print(len(stack))

    elif line[0] == "empty":
        if len(stack) == 0:
            print(1)
        else:
            print(0)

    elif line[0] == "top":
        if len(stack) > 0:
            print(stack[-1])
        else:
            print(-1)

 클래스로 묶어서 코딩하면 더 깔끔한 코드가 될 것 같다. 

처음에 pop에서 stack.pop()대신 stack.remove(stack[-1])라는 코드를 썼었는데 오류가 났다.

이는 스택에 똑같은 숫자가 여러 개 들어있으면 remove가 스택의 맨 위가 아닌 다른 숫자를 지워버릴 수 있기 때문이라고 한다..! 

 

<Java>

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        Stack stack = new Stack<Integer>();
        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine(), " ");
            String command = st.nextToken();

            if (command.equals("push"))
                stack.push(Integer.parseInt(st.nextToken()));

            else if (command.equals("pop"))
                if (stack.size() == 0)
                    sb.append((-1) + "\n");
                else
                    sb.append(stack.pop() + "\n");

            else if (command.equals("size"))
                sb.append(stack.size() + "\n");

            else if (command.equals("empty"))
                if (stack.size() == 0)
                    sb.append(1 + "\n");
                else
                    sb.append(0 + "\n");

            else if (command.equals("top"))
                if (stack.size() == 0)
                    sb.append((-1) + "\n");
                else
                    sb.append(stack.peek() + "\n");

        }
        System.out.println(sb);
    }
}

 

 

▼ 링크

https://www.acmicpc.net/problem/10828

10828번: 스택