테스트 케이스 하나가 오답인데 어떤 부분이 문제인지 모르겠습니다ㅠ

인프런 커뮤니티 질문&답변

tjrwls08088

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

작성

143

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;


class Main {
    public static void main(String[] args) throws IOException {
        Main T = new Main();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        String str1 = br.readLine();
        String str2 = br.readLine();
        System.out.println(T.solution(str1, str2));
    }

    public static String solution(String str1, String str2) {

        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);
        }
        String answer = "";
        for (char x : map1.keySet()) {
            if (map1.containsKey(x) == map2.containsKey(x)) {
                if (map1.get(x) == map2.get(x)) answer = "YES";
                else answer = "NO";
            }
        }

        return answer;
    }
}

테스트케이스 하나를 통과하지 못하는데 제가 생각하지 못한 부분이 있는지 궁금합니다!

코드가 효율적이지 않을 수는 있으나 저렇게 해도 풀리지 않을까? 해서 작성해봤는데.. 뭐가 문제일까요

테스트 케이스

ABCDqtqtqEFqGHIJKLMNOPQRSTUVWetagdgXYabcdefghijklmnopqrstuwxyz

aBcdewrqwtqFghIJklMnOpqrsTuegagaeVxyYAbCDEfGHijKLmNoPQRStUwWXz

java 코딩-테스트 hashmap

답변 1

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다.

tjrwls08088님 코드에서 발생할 수 있는 문제 중 하나는 모든 키에 대해 같은 값을 가지는 경우에만 “YES”가 아닌 경우 “NO”를 바로 리턴하지 않고 계속 루프를 도는 점입니다. 두 문자열이 다르다는 것이 하나의 문자에서라도 확인되면 즉시 “NO”를 반환하고 루프를 종료해야 합니다. 현재 로직에서는 마지막에 확인한 문자가 동일하다면 “YES”로 값을 덮어씌우게 됩니다.

해결책으로는 forEach 루프 내에서 map2.containsKey(x) 체크를 하며, map1과 map2에서 같은 key에 대한 value가 다른 경우 바로 “NO”를 반환하게 수정할 수 있습니다. 모든 키를 확인한 후에 문제가 없으면 “YES”를 반환합니다.

이렇게 코드를 수정해 볼 수 있겠습니다:

public static String solution(String str1, String str2) {
    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 (!map2.containsKey(x) || !map1.get(x).equals(map2.get(x))) {
            return "NO";
        }
    }

    // 만약 map2에도 map1에 없는 추가 문자가 있다면, "NO" 반환
    for (char x : map2.keySet()) {
        if (!map1.containsKey(x)) {
            return "NO";
        }
    }

    return "YES";
}

map2에도 map1에 없는 키가 있는지도 확인해야 올바른 비교를 할 수 있습니다. 위 코드의 수정을 통해 문제를 해결해보시길 바랍니다.

tjrwls08088

작성한 질문수

전체 Q&A

질문하기