[스택] 백준 #3986 좋은 단어 - Python

🗂️ 문제

링크: https://www.acmicpc.net/problem/3986

3986번: 좋은 단어

 

💡 접근법

단어의 알파벳 하나하나를 스택의 마지막 요소와 비교한다.

이때 같을 경우 스택의 마지막 요소를 pop하고, 같지 않을 경우 알파벳을 스택에 append한다.

알파벳을 모두 순회하고 결과적으로 스택이 비어있다면 좋은 단어로 판단할 수 있다.

 

😎 내 코드

import sys

N = int(sys.stdin.readline())
words = []
for _ in range(N):
    words.append(sys.stdin.readline().rstrip())

cnt = 0
for word in words:
    stack = []
    for w in word:
        if stack and stack[-1] == w:
            stack.pop()
        else:
            stack.append(w)
    
    if not stack:
        cnt += 1

print(cnt)

 

✅ 정답 확인