[BOJ] 3986 - 좋은 단어 JAVA

www.acmicpc.net/problem/3986

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

 

 

 

 

 

 

풀이 .

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());

        int ans = 0;
        while(n-- > 0) {
            Stack<Character> stk = new Stack<>();
            String str = br.readLine();
            boolean isGood = true;

            int len = str.length();
            for(int i = 0; i < len; i++) {
                char ch = str.charAt(i);
                if(stk.isEmpty()) {
                    stk.push(ch);
                }else {
                    if(stk.peek() == ch) {
                        stk.pop();
                    }else {
                        stk.push(ch);
                    }
                }
            }
            if(stk.isEmpty()) ans += 1;
        }
        System.out.println(ans);
    }
}

 

문제를 한 번에 이해하기가 어려웠다.

 

한 문자의 위로 선을 뽑아서 다른 위치에 있는 다른 문자와 연결하는 식으로 모든 문자를 연결했을 때 이 선들이 겹치지 않을 수 있으면 '좋은 단어'이다.

 

주어지는 문자열은 A, B로만 이루어져 있는데 이 둘이 연속적으로 교차하지만 않으면 된다.

 

스택을 통해 처리할 수 있다.

 

1. 비어있으면 그냥 넣는다.

2. 비어있지 않으면 peek()가 현재 문자와 같은지 비교한다.

    2.1. 같으면 현재 문자와 peek() 둘 다 제거

    2.2. 다르면 그냥 넣는다.

3. 모든 문자에 대해서 위 연산 수행 후 스택이 비어있으면 '좋은 단어'이다.

 

복잡하게 꼬여있는 단어의 경우 '좋은 단어'여도 검사하지 못 할 수도 있지 않을까 싶었는데 3번에서 검사를 하기 때문에 문제가 없다.