백준 스택 코틀린

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

10828번: 스택

스택 성공

시간 제한메모리 제한제출정답맞은 사람정답 비율

0.5 초 (추가 시간 없음)

256 MB

119374

45372

32662

38.207%

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1 

14

push 1

push 2

top

size

empty

pop

pop

pop

size

empty

pop

push 3

empty

top

예제 출력 1 

2 2 0 2 1 -1 0 1 -1 0 3

 

 

---

이 문제는 그냥 스택을 이용해서 반복문 돌리면 되는 문제입니다!

뭐 쉬운 문제라 설명할 건 딱 히 없을거 같은대 

하나 이야기 하고싶은게 있다면 

제가 처음에 if가 아닌 when() 문을 썼는대 

when 을 쓰게 될 경우 시간 초과가 나오고 if else문을 쓰면 시간초과가 뜨지않습니다! 

코테 문제 풀때는 when 문이 아닌 if else 문을 써야 시간의 효율이 더 좋다는 것을 알 수 있습니다!

 

import java.util.*
fun main(){
    var stack = Stack<Int>()
    val num = readLine()?.toInt()

    for(i in 0 until num!!){
        var str = readLine()
        var array = str?.split(" ")
        if(array?.get(0).equals("push")){
            stack.push(array?.get(1)?.toInt())
        }else if(array?.get(0).equals("pop")){
            if(stack.empty()){
                println(-1)
            }else{
                println(stack.pop())
            }
        }else if(array?.get(0).equals("size")){
            println(stack.size)
        }else if(array?.get(0).equals("empty")){
            if(stack.empty()){
                println(1)
            }else{
                println(0)
            }
        }else if(array?.get(0).equals("top")){
            if(stack.empty()){
                println(-1)
            }else{
                println(stack.peek())
            }
        }
    }
}