[프로그래머스] 코딩테스트 연습 > 연습문제 > 대충 만든 자판
문제 설명
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.
같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.
이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.
1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.
단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.
제한사항
- 1 ≤ keymap의 길이 ≤ 100
- 1 ≤ keymap의 원소의 길이 ≤ 100
- keymap[i]는 i + 1번 키를 눌렀을 때 순서대로 바뀌는 문자를 의미합니다.
- 예를 들어 keymap[0] = "ABACD" 인 경우 1번 키를 한 번 누르면 A, 두 번 누르면 B, 세 번 누르면 A 가 됩니다.
- keymap의 원소의 길이는 서로 다를 수 있습니다.
- keymap의 원소는 알파벳 대문자로만 이루어져 있습니다.
- 1 ≤ targets의 길이 ≤ 100
- 1 ≤ targets의 원소의 길이 ≤ 100
- targets의 원소는 알파벳 대문자로만 이루어져 있습니다.
입출력 예
| keymap | targets | result |
|---|---|---|
| ["ABACD", "BCEFD"] | ["ABCD","AABB"] | [9, 4] |
| ["AA"] | ["B"] | [-1] |
| ["AGZ", "BSSS"] | ["ASA","BGZ"] | [4, 6] |
풀이
내가 푼 코드가 살짝 난잡하긴 하다.
더 개선할 수 있을 거 같다는 생각이 든다.
keymap을 ArrayList를 사용하여 저장하였다.
ArrayList에는 HashMap을 저장하여, 하나의 키에 할당된 문자들을 저장하였다.
key = 할당된 문자열
value = 할당된 문자열이 해당 키의 몇 번째인지
하나의 키에 똑같은 문자가 여러번 할당되어 있더라도, 키를 누를 최소 횟수를 구하기 때문에 가장 앞 순서를 value에 저장하면 된다.
다음으로 for문을 돌면서
문자열을 입력하기 위해 눌러야하는 문자를
keymap에서 검색하여 그 중 최소로 누르는 횟수를 저장한다.
문자 하나하나에 대해 도출한 최소 횟수를 모두 합한다.
코드
import java.util.*;
class Solution {
public int[] solution(String[] keymap, String[] targets) {
int[] answer = new int[targets.length];
// keymap 정보 저장
ArrayList<HashMap<Character, Integer>> keys = new ArrayList<>();
for (int i = 0; i < keymap.length; i++) {
HashMap<Character, Integer> map = new HashMap<>();
for (int j = 0; j < keymap[i].length(); j++) {
if (!map.containsKey(keymap[i].charAt(j))) { // 키가 이미 저장되어 있다면, 원래 저장된 키가 최소 횟수로 키를 누름
map.put(keymap[i].charAt(j), j + 1);
}
}
keys.add(map);
}
// targets을 최소로 누른 값 구하기
for (int i = 0; i < targets.length; i++) {
int totcnt = 0;
for (int j = 0; j < targets[i].length(); j++) {
boolean isFailure = true;
int mincnt = Integer.MAX_VALUE;
for (int k = 0; k < keys.size(); k++) {
if (keys.get(k).containsKey(targets[i].charAt(j))) {
if (keys.get(k).get(targets[i].charAt(j)) < mincnt) {
mincnt = keys.get(k).get(targets[i].charAt(j));
isFailure = false;
}
}
}
if (isFailure) {
totcnt = -1;
break;
} else {
totcnt += mincnt;
}
}
answer[i] = totcnt;
}
return answer;
}
}