1. Problem 📃

📚 출처 - 10828 - 스택

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

명령은 총 다섯 가지이다.

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

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

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

입출력 예

	예제 입력	예제 출력
--	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
--	7 pop top push 123 top pop top pop	-1 -1 123 123 -1 -1

2. Logic 👨‍🏫

문제에 주어진 그대로 구현하면 된다.
여기서 push를 제외한 나머지 단어는 한 단어이기 때문에 그대로 구현하면 되지만, push와 같은 경우 ex) push 1과 같이 되어 있어 숫자를 떼어내야 한다는 점만 유의하면 이번 문제는 쉽게 해결할 수 있다.

3. Code 💻

1. 내가 푼 코드 😁

import sys
input = sys.stdin.readline

def check(stack, word):
    if word == 'top':
        if len(stack) != 0 and stack[-1]:
            print(stack[-1])
        else:
            print(-1)
    elif word == 'size':
        print(len(stack))
    elif word == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif word == 'pop':
        if len(stack) > 0:
            print(stack.pop())
        else:
            print(-1)
    else:
        tmp = int(word.split(' ')[1])
        stack.append(tmp)

N = int(input().strip())
stack = []

for i in range(N):
    word = input().strip()
    check(stack, word)

4. Feedback 📚

다만 이 문제에서 시간이 오래 걸렸던 부분이 있었다. 로직의 문제가 아니라 input()과 sys.stdin.readline의 차이 때문에 오류가 발생해 오래 걸렸는데 이 부분을 짚고 넘어가려한다.

1. input() vs sys.stdin.readline()

공통점

• 둘 다 입력 받은 값을 출력한다는 공통점이 있습니다.

차이점

input()

• 파라미터로 Prompt Message를 받아서 출력합니다.
  (for문을 사용할 시 prompt message를 처리하는 부분에서 느려질 수 있습니다.)
• 개행문자 \n를 삭제해서 반환해줍니다.

sys.stdin.readline()

• sys 모듈은 파이썬 인터페이스를 제공하는 변수와 함수를 직접 제어할 수 있도록 합니다.
• sys.stdin.readline은 개행문자 \n를 삭제하지 않기 때문에 형변환을 거쳐서 사용해주어야 합니다.

위의 문제에서는 개행문자가 삭제가 되지 않아 top를 입력했음에도 불구하고 마지막 else 부분으로 들어가 헤맸습니다. 그래서 sys.stdin.readline이 개행을 포함해 반환한다는 사실을 알았고 strip()을 사용해 공백을 제거해 주었습니다.

2. strip()

Python의 String 의 내장함수를 활용한 문자제거

• strip([chars]) : 인자로 전달된 문자를 String의 왼쪽과 오른쪽에서 제거합니다.
• lstrip([chars]) : 인자로 전달된 문자를 String의 왼쪽에서 제거합니다.
• rstrip([chars]) : 인자로 전달된 문자를 String의 오른쪽에서 제거합니다.

인자를 전달하지 않을 수도 있으며, 인자를 전달하지 않으면 String에서 공백을 제거합니다.

공백(white space) 제거

text = '   this is gold   '
print('[' + text.rstrip() + ']')
print('[' + text.lstrip() + ']')
print('[' + text.strip() + ']')

-------------------------------------------
# result
[   this is gold]
[this is gold   ]
[this is gold]

동일한 문자 제거

인자로 전달한 한 문자와 동일한 개체를 모두 제거, 동일하지 않은 문자가 나올 때까지 제거한다

text = 'aaaafineappleaaaaa'
print(text.rstrip('a'))
print(text.lstrip('a'))
print(text.strip('a'))

-------------------------------------------
# result
[aaaafineapple]
[fineappleaaaaa]
[fineapple]

중간에 있는 a에는 영향을 미치지 않음

여러 문자 제거

인자로 여러 문자를 전달하면 그 문자들과 동일한 것들을 모두 제거, 동일하지 않은 문자가 나올 때까지 제거한다.

text = "rrrrr865.....pretty....rr,,,,"
print(text.lstrip(',865.r'))
print(text.rstrip(',865.r'))
print(text.strip(',865.r'))
print(text.strip(',568.r'))

-------------------------------------------
# result
[pretty....rr,,,,]
[rrrrr865.....pretty]
[pretty]
[pretty]

위와 동일하게 중간에 있는 문자에 대해서는 영향을 미치지않는다.

• 또한 인자를 하나의 독립적인 개체로 보기떄문에 865를 8,6,5 따로따로 인자로 본것 이므로 568 을 입력하였을때도 동일한 결과과 나온다.
• 공백 또한 하나의 요소로 인정하기 때문에 strip(',8 65.r')는 strip(',865.r')이라는 것이랑 비교할때 공백이라는 요소가 들어간 것으로 보면 된다.