[JAVA]백준 3986번 문제 풀이

# 문제 출제 사이트

https://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을 넘지 않는다.

 

# 출력

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

 

 

# 제출한 소스코드

 

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 result = 0;
        for (int i = 0; i < n; i++) {
            String tempStr = br.readLine();

            Stack<Character> stack = new Stack<>();
            int tempNum = 0;
            for (int j = 0; j < tempStr.length(); j++) {
                char tempChar = tempStr.charAt(j);
                char top = ' ';
                if (!stack.empty()) {
                    top = stack.peek();
                }

                if (top == tempChar) {

                    if (j < tempStr.length() - 1 && tempStr.charAt(j + 1) == tempChar) {
                        break;
                    }
                    result++;
                    break;
                }
                stack.push(tempChar);
            }
        }
        System.out.println(result);
    }
}

• 예시에 나오는것만 맞출려고 하다보니 코드를 이렇게 작성했다.
• 문장 내에 인접한 글자가 한 쌍이라도 같은 문자이면 좋은 문자인줄 알았다.
  
• 입력 값으로 AAAA 들어온다고 하면 좋은 단어의 수에 +1 를 못하고 끝나는 로직이라 틀린 답이다.

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 result = 0;
        for (int i = 0; i < n; i++) {
            String tempStr = br.readLine();

            Stack<Character> stack = new Stack<>();
            for (int j = 0; j < tempStr.length(); j++) {
                char tempChar = tempStr.charAt(j);
                char top = ' ';
                if (!stack.empty()) {
                    top = stack.peek();
                }

                if (top == tempChar) {
                    stack.pop();
                } else {
                    stack.push(tempChar);
                }
            }
            if (stack.empty()) result++;
        }
        System.out.println(result);
    }
}

•  j번째 있는 글자와 stack에 top에 있는 글자가 같으면 pop을 해준다.
• 여기서 고려해야 할 건 입력된 문장을 아치형 곡선 형태로 만들어 좌 우 같은 패턴을 이루면 좋은 단어라는 것이다.
• 입력된 문장의 문자만큼 for문을 돌리는 부분이 핵심 로직이다.
  1. j번째 문자를 char자료형 tempChar 변수에 할당한다.
  2. stack이 비어있지 않다면 맨 위에 있는 값(top)을 char자료형인 top에 할당한다.
  3.  j번째 문자를 stack에 push 하기 전에 top에 있는 문자와 같은지 확인한다.
    => j번째 문자와 stack의 top의 문자가 같으면 pop을 한다.
    => 같지 않으면 j번째 문자를 stack에 push한다.
  4. 반복문이 종료되고 stack이 비어있다면 result 값에 1을 더한다(좋은 단어의 수 + 1).