[Coding Test] 백준 #10828 스택

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

10828번: 스택

 

시도 1 - 정답

import sys
n = int(sys.stdin.readline())
arr = []
for i in range(n):
    comm = sys.stdin.readline().strip()
    
    if "push" in comm:
        
        c,num = comm.split()
        num = int(num)
        arr.append(num)
        
    elif comm=="top":
        if len(arr)==0:
            print(-1)
        else:
            print(arr[len(arr)-1])
    
    elif comm=="pop":
        if len(arr)!=0:
            print(arr.pop())
        else:
            print(-1)    
        
    elif comm=="empty":
        if len(arr)==0:
            print(1)
        else:
            print(0)
    elif comm =="size":
       print(len(arr))

 

처음에 input()으로 했더니 시간초과 떠서 sys.stdin.readline()으로 바꿨더니 틀렸습니다 뜸 ...

보니까 sys.stdin.readline()은 개행도 해줘서 strip()으로 개행 없애고 다시 넣었더니 통과!!

 

근데 파이썬은 -1번째 인덱스가 마지막 값이므로 이렇게 해서 풀면 len() 메소드를 쓰는 것보다 시간이 더 짧게 나오지 않을까 싶어서 수정해봤다.

---

시도 2 - 정답

import sys
n = int(sys.stdin.readline())
arr = []
for i in range(n):
    comm = sys.stdin.readline().strip()
    
    if "push" in comm:
        
        c,num = comm.split()
        
        arr.append(int(num))
        
    elif comm=="top":
        if arr:
            print(arr[-1])
            
        else:
            print(-1)
    
    elif comm=="pop":
        if arr:
            print(arr.pop())
        else:
            print(-1)    
        
    elif comm=="empty":
        if arr:
            print(0)
        else:
            print(1)
    elif comm =="size":
       print(len(arr))

 

 if len(arr) !=0:  => if arr: 

print(arr[len(arr)-1]) => print(arr[-1])

 

시간은 똑같았지만 그래도 더 간결하게 쓸 수 있으니 수정 버전으로 쓰도록 해야겠다.