C_C++

C++ - 백준 3986 좋은 단어

ncode 2025. 2. 16. 09:14
728x90
반응형
/* 백준 3986 좋은 단어

* 문제

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

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

* 입력

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

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

* 출력

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

* 예제 입력 1

3
ABAB
AABB
ABBA

* 예제 출력 1

2

* 예제 입력 2

3
AAA
AA
AB

* 예제 출력 2

1

* 예제 입력 3

1
ABBABB

* 예제 출력 3

1

*/

/* 풀이

주어진 문자열의 각 문자들을 순서대로 반복하면서, 
stack의 가장 위에 있는 요소가 같은 문자이면 pop, 다른 문자이면 push 하면서 
최종적으로 stack 이 empty 가 되면 좋은 단어로 판정하면 된다.

예) ABAB
     _
    | |
    ABAB  --> 교차하지 않고 아치를 만들 방법이 없음


    Stack A         : 처음 문자는 push
          AB        : stack 의 top 이 A 이고 현재 문자가 B 이므로 push
          ABA       : stack 의 top 이 B 이고 현재 문자가 A 이므로 push
          ABAB      : 마찬가지로 push

    --> empty 가 아니므로 "좋은 단어" 가 아님

    ABBA

    Stack A         : 처음 문자 push
          AB        : stack 의 top 이 A, 현재 문자가 B 이므로 push
          A         : 현재 문자가 B 인데 top 이 B 이므로 pop 한다
          -         : 현재 문자가 A 인데 top 이 A 이므로 pop

    --> empty 이므로 "좋은 단어" 임

*/

#include <algorithm>
#include <iostream>
#include <stack>
#include <string>

using namespace std;

int main() {
	int n;
	cin >> n;

	int count = 0;

	while (n-- > 0) {
		string str;
		cin >> str;

		stack<char> s;

		for (auto ch : str) {
			if (s.empty()) {
				s.push(ch);
				continue;
			}

			if (s.top() == ch)
				s.pop();
			else
				s.push(ch);
		}

		if (s.empty())
			++count;
	}

	cout << count << '\n';

	return 0;
}
728x90
반응형