[PS] 인프런 강의 - HashMap, HashSet, TreeSet 2. 아나그램(해쉬)

 

문제

Anagram이란 두 문자열이 알파벳의 나열 순서를 다르지만 그 구성이 일치하면 두 단어는 아나그램이라고 합니다.
예를 들면 AbaAeCe 와 baeeACA 는 알파벳을 나열 순서는 다르지만 그 구성을 살펴보면
A(2), a(1), b(1), C(1), e(2)로 알파벳과 그 개수가 모두 일치합니다. 즉 어느 한 단어를 재배열하면 상대편 단어가 될 수 있는 것을 아나그램이라 합니다.
길이가 같은 두 개의 단어가 주어지면 두 단어가 아나그램인지 판별하는 프로그램을 작성하세요. 아나그램 판별시 대소문자가 구분됩니다.

 

▣ 입력설명
첫 줄에 첫 번째 단어가 입력되고, 두 번째 줄에 두 번째 단어가 입력됩니다.
단어의 길이는 100을 넘지 않습니다.

 

▣ 출력설명
두 단어가 아나그램이면 “YES"를 출력하고, 아니면 ”NO"를 출력합니다.

 

▣ 입력예제 1
AbaAeCe
baeeACA

 

▣ 출력예제 1
YES

 

▣ 입력예제 2
abaCC
Caaab

 

▣ 출력예제 2
NO

 

풀이

처음엔 그냥 HashMap을 2개 두고 비교해서 풀었다...

 

강상님은 푼 방법은 이렇다!!

 

1개를 두고 문자열1에서 받은 값으로 key-value 쌍을 추가 한 다음, 문자열2를 체크할 때는 HashMap에 있는 문자면 HashMap의 value를 -1 한다.

 

이렇게 하면 같은 키가 같은 개수만큼 있을 경우 반복문이 끝나면 그 키의 value는 0이 될것이다.

 

문자열 2를 순회할 때 해당 키를 가지고 있지 않거나, 키의 value가 0이면 이번에 개수를 초과했다는 뜻이므로 "NO"를 출력한다.

 

코드

1) 처음에 푼 방식

import java.util.HashMap;
import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		String s1 = in.nextLine();
		String s2 = in.nextLine();;
		String answer = "YES";
		//알파벳과 그 개수가 모두 일치하면 아나그램. 대소문자 구분됨
		HashMap<Character,Integer> ana1 = new HashMap<>();
		HashMap<Character,Integer> ana2 = new HashMap<>();
		for(char x : s1.toCharArray()) {
			ana1.put(x, ana1.getOrDefault(x, 0)+1);
		}
		for(char x : s2.toCharArray()) {
			ana2.put(x, ana2.getOrDefault(x, 0)+1);
		}
		
		for(char key : ana1.keySet()) {
			if(ana1.get(key) != ana2.get(key)) {
				answer = "NO"; break;
			}
			else if(ana2.containsKey(key) == false) {
				answer = "NO"; break;
			}
		}
		
		System.out.print(answer);
	}
}

2) 강사님 풀이

import java.util.HashMap;
import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		
		String s1 = in.nextLine();
		String s2 = in.nextLine();;
		String answer = "YES";
		//알파벳과 그 개수가 모두 일치하면 아나그램. 대소문자 구분됨
		HashMap<Character,Integer> map = new HashMap<>();

		for(char x : s1.toCharArray()) {
			map.put(x, map.getOrDefault(x, 0)+1);
		}
		for(char x : s2.toCharArray()) {
			if(!map.containsKey(x) || map.get(x)==0) {
				answer = "NO";
				break;
			}
			map.put(x, map.get(x)-1);
		}
		
		System.out.print(answer);
	}
}