[프로그래머스/Lv.1] 숫자 짝꿍 - Java

📌 문제

프로그래머스

📌 풀이

두 개의 정수의 임의의 자리에서 공통으로 나타나는 정수를 이용해 만들 수 있는 가장 큰 정수를 반환해야 하는 문제였다. 전체적인 흐름은 다음과 같이 정리할 수 있다.

• 가장 먼저 X와 Y에서 공통된 숫자를 찾기 위해 X, Y를 돌면서 저장할 HashMap을 두 개 만든다.
  • 각각의 HashMap은 각 자리의 정수를 key로 갖고, 해당 정수의 개수를 value로 갖는다.
• 최종적으로 만들어진 HashMap을 모두 순회하면서 각 map의 key로 같은 숫자가 존재한다면(짝지을 수 있다면)
  • 해당 숫자의 개수가 더 작은 쪽을 int형 변수 n에 저장한다.
  • 해당 숫자를 StringBuilder에 n번 반복해 추가한다.
• StringBuilder를 내림차순으로 정렬한다.
• StringBuilder가 비어있다면 "-1"을 반환한다.
• StringBuilder의 0번째 요소가 '0'이라면 "0"을 반환한다.
• 모두 해당되지 않으면 StringBuilder.toString()을 반환한다.

HashMap에 key와 value를 저장하는 것부터 시작한다. X와 Y를 char 배열로 만들어 순회하였다. map.getOrDefault() 메서드를 이용해 각 key에 대한 value가 설정되어있지 않으면 default 값으로 0을 넣도록 하고, value가 이미 있다면 그 value를 그대로 불러오도록 했다. 어떻게 불러오든 + 1을 해서 해당 key의 개수를 세도록 했다.

Map<Character, Integer> mapX = new HashMap<>();
for (char c : X.toCharArray()) {
    mapX.put(c, mapX.getOrDefault(c, 0) + 1);
}
        
Map<Character, Integer> mapY = new HashMap<>();
for (char c : Y.toCharArray()) {
    mapY.put(c, mapY.getOrDefault(c, 0) + 1);
}

 

그리고나서 HashMap의 keySet()을 하나씩 순회하면서 mapX의 key와 mapY의 key가 같을 경우 그 개수만큼 StringBuilder에 추가하도록 했다. 여기서 개수는 두 개의 map중 더 작은 수를 말한다.

StringBuilder answer = new StringBuilder();
for (Character keyX : mapX.keySet()) {
    for (Character keyY : mapY.keySet()) {
        if (keyX == keyY) {
            int n  = (int) Math.min(mapX.get(keyX), mapY.get(keyY));
            answer.append(keyX.toString().repeat(n));
        }
    }
}

 

마지막으로 가장 큰 정수를 만들어야 하므로 내림차순으로 정렬한 후 StringBuilder가 비어있을 때와 0으로만 구성되어 있을 때 조건 처리를 해준다.

answer.reverse();
if (answer.toString().isEmpty()) return "-1";
if (answer.toString().charAt(0) == '0') return "0";

return answer.toString();

📌 전체 코드

import java.util.*;

class Solution {
    public String solution(String X, String Y) {
        Map<Character, Integer> mapX = new HashMap<>();
        for (char c : X.toCharArray()) {
            mapX.put(c, mapX.getOrDefault(c, 0) + 1);
        }
        
        Map<Character, Integer> mapY = new HashMap<>();
        for (char c : Y.toCharArray()) {
            mapY.put(c, mapY.getOrDefault(c, 0) + 1);
        }
        
        StringBuilder answer = new StringBuilder();
        for (Character keyX : mapX.keySet()) {
            for (Character keyY : mapY.keySet()) {
                if (keyX == keyY) {
                    int n  = (int) Math.min(mapX.get(keyX), mapY.get(keyY));
                    answer.append(keyX.toString().repeat(n));
                }
            }
        }
        
        answer.reverse();
        if (answer.toString().isEmpty()) return "-1";
        if (answer.toString().charAt(0) == '0') return "0";
        
        return answer.toString();
    }
}