백준[10828.스택] | Python

✏️ 문제

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

10828번: 스택

 

✏️ 풀이방법

1. 입력하는 값이 많기 때문에 input()이 아닌 sys.stdin.readline()을 통해 입력을 받는다.

2. if-elif문을 사용하여 FILO(First In Last Out)으로 작성한다.

 

 

📌 코드

import sys
n = int(sys.stdin.readline())
stack = []
for i in range(n):
    s = sys.stdin.readline().split()
    if s[0] == 'push':
        stack.append(s[1])
    elif s[0] == 'pop':
        if len(stack) > 0:
            print(stack.pop())
        else:
            print(-1)
    elif s[0] == 'size':
        print(len(stack))
    elif s[0] == 'empty':
        if len(stack) > 0:
            print(0)
        else:
            print(1)
    elif s[0] == 'top':
        if len(stack) > 0:
            print(stack[-1])
        else:
            print(-1)

 

 

📌 결과

#입력
14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top


#출력
2
2
0
2
1
-1
0
1
-1
0
3

 

 

🤔 시행착오.1

import sys
n = int(sys.stdin.readline())
stack = []
for i in range(n):
    s = sys.stdin.readline().split()
    if s[0] == 'push':
        stack.append(s[1])
    elif s[0] == 'pop':
        if len(stack) > 0:
            a = stack[-1]
            stack.remove(stack[-1])
            print(a)
        else:
            print(-1)
    elif s[0] == 'size':
        print(len(stack))
    elif s[0] == 'empty':
        if len(stack) > 0:
            print(0)
        else:
            print(1)
    elif s[0] == 'top':
        if len(stack) > 0:
            print(stack[-1])
        else:
            print(-1)

- 출력은 예시와 같으나 틀린 답

- remove는 첫 번째 원소부터 탐색하여 가장 먼저 등장하는 값을 제거하기 때문에 인덱스로 제거하는 것이 아니라 stack[-1]로 작성했지만 위치가 아닌 값이 제거되서 오답이 된다.

 

 

 

📝 메모

- pop 메소드는 인덱스로 값을 제거하고, remove는 값을 통해서 먼저 등장하는 값으로 제거한다.

- 입력 값이 많을 때는 input() 내장 함수가 아닌 sys모듈의 sys.stdin.readline()을 사용하면 시간초과가 되지 않는다.

 

📒 다시보기

[Python] 05. 파이썬 문자열/리스트 관련 메소드