🔒 문제 설명
이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에 깨고 말았다. 안타깝게도 자는 동안 키보드가 잘못 눌려서 보고서의 모든 글자가 A와 B로 바뀌어 버렸다! 그래서 평석이는 보고서 작성을 때려치우고 보고서에서 '좋은 단어'나 세보기로 마음 먹었다.
평석이는 단어 위로 아치형 곡선을 그어 같은 글자끼리(A는 A끼리, B는 B끼리) 쌍을 짓기로 하였다. 만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을수 있다면, 그 단어는 '좋은 단어'이다. 평석이가 '좋은 단어' 개수를 세는 것을 도와주자.
🔓 소스 코드
import sys
input = sys.stdin.readline
N = int(input())
count = 0
for _ in range(N):
s = input().rstrip()
stack = []
for i in range(len(s)):
if stack and s[i] == stack[-1]:
stack.pop()
else:
stack.append(s[i])
if not stack:
count += 1
print(count)🔑 해결 방법
처음에는 문제를 이해하는데 오래걸렸는데 그림으로 그리면서 이해를 했다.
만약, ABBA 라면 A-A, B-B 연결을 할 때 교차하지 않는다. 그러나 ABAB는 A-A, B-B 연결할 때 교차하기 때문에 카운트를 하지 않는다. 이런식으로 문제를 풀면 스택을 사용해서 pop과 append를 해주면 된다.
만약 stack이 있고 스택의 마지막 값과 지금 입력된 값이 같으면 pop을 해주고 아니라면 append를 해준다. 그 모든 과정을 하고 stack이 없다면! 이 말은 즉 교차 안되고 모든 값이 삭제 됐을 때 count를 +1를 해준다.
