[백준] 3986번 좋은단어

[백준] 3986번 좋은단어 - 파이썬

bled 2021. 6. 2. 20:00

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

3986번: 좋은 단어

이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에

www.acmicpc.net

[코드]

import sys

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

answer = 0

for word in words:
    stack = []
    for char in word:
        if stack:
            if stack[-1] == 'A':
                if char == 'A':
                    stack.pop()
                else:  # 'B'
                    stack.append(char)
            else:  # 'B'
                if char == 'B':
                    stack.pop()
                else:  # 'A'
                    stack.append(char)
        else:
            stack.append(char)
    if not stack:
        answer += 1

print(answer)