[99클럽 코테 스터디 9일차 TIL] 백준 3986번 좋은 단어 해설 및 풀이 (Java, Python)

 

백준 좋은 단어

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

출력

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

---

 

문제 유형

 

자료구조

스택

 

 

풀이 방법 도출

 

이 문제는 짝지을 수 있는 '좋은 단어'의 개수를 찾는 문제입니다.

 

여기서 좋은 단어란?

• 같은 글자끼리 정확히 하나씩 짝지을 수 있어야 함 (모두 짝지어짐).
• 짝지을 때 선이 교차하지 않아야 함.
• 즉, 스택 구조처럼 순서에 맞게 짝지어야 함.

 

1. 각 단어를 순회하면서 스택을 사용해 짝을 맞춰본다.
2. 스택이 비어있다면 좋은 단어이다. 짝을 정확히 맞춘 경우는 push, pop이 완벽하게 이루어져서 스택이 비게 된다.
3. 짝이 맞지 않으면 스택에 값이 남는다 → 좋은 단어가 아님.

 

 

시간 복잡도

• 각 단어의 길이를 L이라고 했을 때, 단어 하나당 최대 O(L)

 

---

핵심 코드 (Java)

 

import java.util.Scanner;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            String str = sc.next();
            Stack<Character> stack = new Stack<>();
            boolean isGoodWord = true;
            for (int j = 0; j < str.length(); j++) {
                char ch = str.charAt(j);
                if (stack.isEmpty() || stack.peek() != ch) {
                    stack.push(ch);
                } else {
                    stack.pop();
                }
            }
            if (stack.isEmpty()) {
                cnt++;
            }
        }
        System.out.println(cnt);
        sc.close();
    }
}

 

핵심 코드 (Python)

 

N = int(input())

# 좋은 단어의 개수를 세기 위한 변수
cnt = 0

for _ in range(N):
    stack = [] # 스택 리스트
    S = list(input()) # 단어 입력
    
    # 주어진 단어에서 한 문자씩 처리
    for i in S:
        # 스택이 비어있으면
        if not len(stack): 
            # 현재 문자를 문자 쌍을 맞추기 위해 스택에 넣음   
            stack.append(i)
        # 스택의 맨 위에 있는 문자와 현재 문자가 같으면
        elif stack[-1] == i:
            # 짝이 맞는 것이므로 스택에서 pop
            stack.pop(-1)
        # 스택의 맨 위에 있는 문자와 현재 문자가 다르면
        else:
            # 짝이 맞지 않으므로 스택에 현재 문자 추가
            stack.append(i)
    
    # 스택이 비어있으면 
    if not len(stack):
        # 모든 문자 쌍이 맞춰졌다는 의미로, 좋은 단어임
        # 따라서 cnt를 1 증가
        cnt += 1

print(cnt)