[Inflearn] 자바 알고리즘 문제풀이 #04-02 2. 아나그램(해쉬)

문제

설명

Anagram이란 두 문자열이 알파벳의 나열 순서를 다르지만 그 구성이 일치하면 두 단어는 아나그램이라고 합니다.

예를 들면 AbaAeCe 와 baeeACA 는 알파벳을 나열 순서는 다르지만 그 구성을 살펴보면 A(2), a(1), b(1), C(1), e(2)로

알파벳과 그 개수가 모두 일치합니다. 즉 어느 한 단어를 재 배열하면 상대편 단어가 될 수 있는 것을 아나그램이라 합니다.

길이가 같은 두 개의 단어가 주어지면 두 단어가 아나그램인지 판별하는 프로그램을 작성하세요. 아나그램 판별시 대소문자가 구분됩니다.

 

입력 

첫 줄에 첫 번째 단어가 입력되고, 두 번째 줄에 두 번째 단어가 입력됩니다.

단어의 길이는 100을 넘지 않습니다.

 

출력 

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

 

 

 

내 풀이

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

public class Main4_2 {
    public String Solution(String str1, String str2) {
        String answer = "NO";
        int com_cnt=0;
        HashMap<Character, Integer> map1 = new HashMap<>();
        HashMap<Character, Integer> map2 = new HashMap<>();
        for(char x : str1.toCharArray()) map1.put(x, map1.getOrDefault(x, 0)+1);
        for(char x : str2.toCharArray()) map2.put(x, map2.getOrDefault(x, 0)+1);
        for(char x : map1.keySet()) {
            if(map1.get(x)==map2.get(x)) com_cnt++;
        }
        if(com_cnt==map1.size()) answer="YES";
        return answer;
    }

    public static void main(String[] args) {
        Main4_2 T = new Main4_2();
        Scanner sc = new Scanner(System.in);
        String str1 = sc.next();
        String str2 = sc.next();
        System.out.println(T.Solution(str1, str2));
    }
}

 

 

 

선생님 풀이

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

public class Main4_2 {
    public String Solution(String s1, String s2) {
        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) // 같은 문자가 없거나 개수가 다른 경우
                return "NO";
            map.put(x, map.get(x)-1);
        }
        return answer;
    }

    public static void main(String[] args) {
        Main4_2 T = new Main4_2();
        Scanner sc = new Scanner(System.in);
        String a = sc.next();
        String b = sc.next();
        System.out.println(T.Solution(a, b));
    }
}

나는 아직 예외를 다루는 코드(예외가 발생했을 시 프로그램이 빨리 종료되게 하는 코드)를 먼저 생각하지 못하는 것 같다.

HashMap 두개를 비교할 때는 한 번은 +, 한 번은 -을 해서 예외가 발생한다면 같지 않다! 예외가 발생하지 않는다면 같은 것이다!

방식이 자주 사용되는 것 같으니 추가로 기억해두자.

 

 

 

결과