[백준 3986] 좋은 단어 (C++)

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

 

문제

---

이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에 깨고 말았다. 안타깝게도 자는 동안 키보드가 잘못 눌려서 보고서의 모든 글자가 A와 B로 바뀌어 버렸다! 그래서 평석이는 보고서 작성을 때려치우고 보고서에서 '좋은 단어'나 세보기로 마음 먹었다.

평석이는 단어 위로 아치형 곡선을 그어 같은 글자끼리(A는 A끼리, B는 B끼리) 쌍을 짓기로 하였다. 만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을수 있다면, 그 단어는 '좋은 단어'이다. 평석이가 '좋은 단어' 개수를 세는 것을 도와주자.

 

입력

---

첫째 줄에 단어의 수 N이 주어진다. (1 ≤ N ≤ 100)

다음 N개 줄에는 A와 B로만 이루어진 단어가 한 줄에 하나씩 주어진다. 단어의 길이는 2와 100,000사이이며, 모든 단어 길이의 합은 1,000,000을 넘지 않는다.

 

출력

---

첫째 줄에 좋은 단어의 수를 출력한다.

 

문제 풀이

---

스택을 이용하는 문제.

단어 위로 아치형 곡선을 그어 같은 글자끼리 쌍을 짓는데 선끼리 교차하지 않게 하려면 이전의 문자들과 쌍을 지을 수 있는지 확인해야 한다. 해당 문제는 스택의 괄호의 쌍을 짓는 문제로, A 대신 (), B 대신 {}로 생각하면 쉽다.

스택의 top 값이 없거나 현재 탐색 중인 인덱스의 값이 스택의 top과 동일하다면 쌍을 지을 수 있다. 현재 탐색 중인 인덱스의 값이 스택과 다르다면 여는 괄호와 역할이 같으므로 스택에 넣는다.

모든 문자를 순회한 뒤, 짝이 제대로 맞았다면 스택에는 남는 값이 없게 된다. 예를 들어 ABBA는 ({})와 같은 형태로, 올바른 형태의 괄호 쌍(=좋은 단어)이 되고 스택의 크기가 0이다. ABAB의 경우 ({)}와 같은 형태로, 아직 {가 닫히지 않았는데 (가 나와 올바르지 않은 형태가 된다.

 

아래는 코드.

#include <iostream>
#include <stack>
using namespace std;

int main()
{
	cin.tie(NULL);
	ios::sync_with_stdio(false);

	int N;
	string str;
	stack<char> s;
	int answer = 0;
	cin >> N;
	for (int i = 0; i < N; i++)
	{
		cin >> str;
		for (int j= 0; j < str.length(); j++)
		{
			if (s.empty())
			{
				s.push(str.at(j));
			}
			else
			{
				if (s.top() == str.at(j))
				{
					s.pop();
				}
				else
				{
					s.push(str.at(j));
				}
			}
		}
		if (s.empty())
		{
			answer++;
		}
		else
		{
			while (!s.empty())
			{
				s.pop();
			}
		}
	}
	cout << answer << "\n";
	return 0;
}