3986번: 좋은 단어
이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에
www.acmicpc.net
문제
이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에 깨고 말았다. 안타깝게도 자는 동안 키보드가 잘못 눌려서 보고서의 모든 글자가 A와 B로 바뀌어 버렸다! 그래서 평석이는 보고서 작성을 때려치우고 보고서에서 '좋은 단어'나 세보기로 마음 먹었다.
평석이는 단어 위로 아치형 곡선을 그어 같은 글자끼리(A는 A끼리, B는 B끼리) 쌍을 짓기로 하였다. 만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을수 있다면, 그 단어는 '좋은 단어'이다. 평석이가 '좋은 단어' 개수를 세는 것을 도와주자.
입력
첫째 줄에 단어의 수 N이 주어진다. (1 ≤ N ≤ 100)
다음 N개 줄에는 A와 B로만 이루어진 단어가 한 줄에 하나씩 주어진다. 단어의 길이는 2와 100,000사이이며, 모든 단어 길이의 합은 1,000,000을 넘지 않는다.
출력
첫째 줄에 좋은 단어의 수를 출력한다.
코드
def chk(w):
stack = []
for i in range(len(w)):
if not stack or stack[-1] != w[i]:
stack.append(w[i])
else:
stack.pop()
return False if stack else True
n = int(input())
ans = 0
for _ in range(n):
word = input()
if chk(word):
ans += 1
print(ans)
chk() 함수는 매개 변수로 받은 w가 '좋은 단어'인지 아닌지 판별하는 함수이다.
좋은 단어는 BAABAA와 같이 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을수 있는 단어이다.
이걸 풀기 위해서는 stack을 사용해야 한다. 괄호로 올바른 괄호인지 판별할 때를 생각하면 편하다. 예를 들어, (())()가 올바른 괄호인지 알아보기 위해서는 stack을 이용하여 '('이 나오면 스택에 넣고, ')'이 나오면 stack의 top을 pop하는 로직을 구상해볼 수 있다.
위에서는 서로 짝지어지는 것이 같은 글자끼리라고 생각하면 된다. 그렇기 때문에 같은 글자가 stack의 top에 있다면 pop하고 그렇지 않고 다른 글자가 top에 있거나 stack이 비었다면 해당 글자를 stack에 push해준다. 만약 for문을 다 돌았는데 stack이 비어있지 않다면 좋은 단어가 될 수 없다는 뜻이다.
'문제풀이 > 자료구조' 카테고리의 다른 글
[Python/파이썬] 백준 12789번 도키도키 간식드리미 (0) | 2024.03.05 |
---|---|
[Python/파이썬] 백준 5397번 키로거 (0) | 2024.01.14 |
[Python/파이썬] 백준 18115번 카드 놓기 (0) | 2023.11.28 |
[Python/파이썬] 백준 22866번 탑 보기 (0) | 2023.11.21 |
[Python/파이썬] 백준 5875번 오타 (0) | 2023.10.27 |