[Python] 백준3986번 : 좋은 단어

💡문제

백준 3986번 문제 링크

🍀풀이

오늘도 역시 처음 문제를 보고 문제 이해를 못했다... 나 문제 이해 못해서 코테 못풀거같은데 ㅋㅋ...
나같은 사람이 좀 있는지 질문게시판에 문제 이해를 위한 정보를 주신분이 계셔서 참고를 했다.

예제를 가지고 해보면 이렇게 되는거다. 이거를 보고 로직을 생각해 봤다.

스택에 문자를 하나씩 넣는데, 가장 최근에 들어간 스택 값이 현재 문자랑 같으면 pop을해서 빼내고, 같지 않으면 append해서 계속 연산을 한다. 한 문자열에 대한 for문이 끝나면 스택에 값이 없으면 좋은 단어이니 count +1을 해준다.

코드

import sys
input = sys.stdin.readline

n = int(input())
count = 0

for _ in range(n):
    command = input().rstrip()
    stack = []
    
    for i in command:
        if len(stack) == 0:
            stack.append(i)
        elif len(stack) != 0 and stack[-1] == i:
            stack.pop()
        elif len(stack) != 0 and stack[-1] != i:
            stack.append(i)
    if len(stack) == 0:
        count += 1
print(count)

코드는 이렇게 짰다! 문제 이해만 되면 쉬운 문제였다.