(Java) 백준 2607번

백준 2607번 실버3 정답률 27.782%

문제

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

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 이하이다.

출력

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

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

public class Main {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		
		int num = Integer.parseInt(br.readLine());
		Map<Character, Integer> core = new HashMap<Character, Integer>();
		int totalCount = 0;
		for(int i = 0; i < num; i++) {
			if(i == 0) {
				char[] arr = br.readLine().toCharArray();
				for(char a : arr) {
					if(core.containsKey(a)) core.put(a, core.get(a) + 1);
					else core.put(a, 1);
				}
			}
			else {
				Map<Character, Integer> diff = new HashMap<Character, Integer>();
				for(Entry<Character, Integer> Entry  : core.entrySet()) {
					diff.put(Entry.getKey(), Entry.getValue());
				}
				String str = br.readLine();
				char[] arr = str.toCharArray();
				for(char a : arr) {
					if(diff.containsKey(a)) diff.put(a, diff.get(a) - 1);
					else diff.put(a, -1);
				}
                boolean isOk = true;
				int isNotZero = 0;
				int sum = 0;
				for(char a : diff.keySet()) {
					if((diff.get(a) == 1 || diff.get(a) == -1) && (isNotZero == 0 || (sum == 1 && diff.get(a) == -1 && isNotZero == 1) || (sum == -1 && diff.get(a) == 1 && isNotZero == 1))) {
                        // 한 문자를 더하거나, 빼거나, 하나의 문자를 다른 문자로 바꾸어 구성이 같은 경우 비슷한 단어
                    }
					else if(diff.get(a) != 0) isOk = false;
                    
                    if(diff.get(a) != 0) isNotZero++;
					sum += diff.get(a);
				}
				if(isOk == true) {
					totalCount++;
				}
			}
		}
		bw.write(totalCount + "");
		bw.flush();
        br.close();
        bw.close();
	}
}

1. 첫 번째 입력받은 문자열의 각 문자별 개수를 Map에 담아놓고( key : 문자, value : 개수 )

2. 두 번째부터 입력받은 문자열과 첫 번째 문자열의 Map과 비교하면서 존재하면 value - 1을, 존재하지 않으면 -1을 새로 put하는 방식으로 차이를 기록해나가고

3. 차이를 기록한 Map을 반복문으로 순회하면서

    (1) 모든 차이가 없으면( 모든 diff 요소의 value가 0이면 ) isOk는 초기화한대로  true

    (2) 한 문자가 차이난다면( diff.get(a) == 1 || diff.get(a) == -1) && (isNotZero == 0) ) isOk는 초기화한대로  true

    (3) 하나의 문자를 다른 문자로 바꾸어 나머지 한 단어와 같은 구성을 갖게 되는 경우( (diff.get(a) == 1 || diff.get(a) == -1) && ((sum == 1 && diff.get(a) == -1 && isNotZero == 1) || (sum == -1 && diff.get(a) == 1 && isNotZero == 1)) ) 이전차이 sum과 diff.get() 결과가 1과 -1로 서로 상쇄되면서 isNotZero 변수로 2개만 서로 자리를 바꾸면 되는 경우 isOk는 초기화한대로  true

    (4) 나머지 경우는 모두 isOk = false

4. isOk = true인 경우의 수만 출력

 

이상입니다. 지금까지 읽어주셔서 감사합니다.