백준 2607 비슷한 단어(Java)

문제 출처 : https://www.acmicpc.net/problem/2607

2607번: 비슷한 단어

문제 설명

영문 알파벳 대문자로 이루어진 두 단어가 다음의 두 가지 조건을 만족하면 같은 구성을 갖는다고 말한다.

1. 두 개의 단어가 같은 종류의 문자로 이루어져 있다.
2. 같은 문자는 같은 개수 만큼 있다.

예를 들어 "DOG"와 "GOD"은 둘 다 'D', 'G', 'O' 세 종류의 문자로 이루어져 있으며 양쪽 모두 'D', 'G', 'O' 가 하나씩 있으므로 이 둘은 같은 구성을 갖는다. 하지만 "GOD"과 "GOOD"의 경우 "GOD"에는 'O'가 하나, "GOOD"에는 'O'가 두 개 있으므로 이 둘은 다른 구성을 갖는다.

두 단어가 같은 구성을 갖는 경우, 또는 한 단어에서 한 문자를 더하거나, 빼거나, 하나의 문자를 다른 문자로 바꾸어 나머지 한 단어와 같은 구성을 갖게 되는 경우에 이들 두 단어를 서로 비슷한 단어라고 한다.

예를 들어 "DOG"와 "GOD"은 같은 구성을 가지므로 이 둘은 비슷한 단어이다. 또한 "GOD"에서 'O'를 하나 추가하면 "GOOD" 과 같은 구성을 갖게 되므로 이 둘 또한 비슷한 단어이다. 하지만 "DOG"에서 하나의 문자를 더하거나, 빼거나, 바꾸어도 "DOLL"과 같은 구성이 되지는 않으므로 "DOG"과 "DOLL"은 비슷한 단어가 아니다.

입력으로 여러 개의 서로 다른 단어가 주어질 때, 첫 번째 단어와 비슷한 단어가 모두 몇 개인지 찾아 출력하는 프로그램을 작성하시오.

입력

첫째 줄에는 단어의 개수가 주어지고 둘째 줄부터는 한 줄에 하나씩 단어가 주어진다. 모든 단어는 영문 알파벳 대문자로 이루어져 있다. 단어의 개수는 100개 이하이며, 각 단어의 길이는 10 이하이다.

출력

입력으로 주어진 첫 번째 단어와 비슷한 단어가 몇 개인지 첫째 줄에 출력한다.

풀이

문제를 풀다보니 생각해야 될 케이스가 많았다.

알파벳 개수인 26개 크기의 int[] 배열을 만들어 첫 번째 문자열 단어에 해당하는 값에 +1씩 해주고
비교 문자열 단어에 해당 배열이 0보다 크면 -1을 해주고 cnt를 1증가 시킨다. 여기선 cnt는 첫 번째 문자열과 같은 단어의 개수를 의미한다.

먼저 크게 첫 번째 단어 길이와 같은 길이, +1, -1로 나눈다.

비슷한 단어의 조건

• 길이가 같은 경우 : 문자열이 둘 다 같거나, 하나만 바뀐 경우 result++
• 길이가 +1인 경우 : 단어가 하나 추가된 것으로 cnt가 첫 번째 문자열 길이와 같으면 result++
• 길이가 -1인 경우 : 단어가 하나 제거된 것으로 비교 문자열의 길이 - 1과 cnt가 같으면 result++

코드

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        String first = br.readLine();
        int result = 0;

        for(int i = 0; i < N - 1; i++) {
            String str = br.readLine();
            int cnt = 0;    // 같은 단어 개수
            int[] word = new int[26];
            for(int j = 0; j < first.length(); j++) {
                word[first.charAt(j) - 'A']++;
            }

            for(int j = 0; j < str.length(); j++) {
                if(word[str.charAt(j) - 'A'] > 0) {
                    cnt++;
                    word[str.charAt(j) - 'A']--;
                }
            }

            // 기준 문자열 길이와 비교 문자열 길이 차이
            // 같은 경우 or 한 글자만 바뀐경우
            if(first.length() == str.length() && (first.length() == cnt || first.length() - 1 == cnt)) {   
                result++;
            } 
            // 비교 문자열이 짧은 경우 (제거)
            else if (first.length() == str.length() - 1 && str.length() - 1 == cnt) {   
                result++;
            } 
            // 비교 문자열이 긴 경우 (추가)
            else if (first.length() == str.length() + 1 && str.length() == cnt) {   
                result++;
            }

        }
        System.out.println(result);

    }
}

결과