프로그래머스 Level 1
🔒 [1차] 비밀지도
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
n
인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.1
, 공백 부분을 0
으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.네오가 프로도의 비상금을 손에 넣을 수 있도록, 비밀지도의 암호를 해독하는 작업을 도와줄 프로그램을 작성하라.
입력으로 지도의 한 변 크기 n
과 2개의 정수 배열 arr1
, arr2
가 들어온다.
n
≦ 16arr1
, arr2
는 길이 n
인 정수 배열로 주어진다.x
를 이진수로 변환했을 때의 길이는 n
이하이다. 즉, 0 ≦ x
≦ 2n - 1을 만족한다.원래의 비밀지도를 해독하여 '#'
, 공백
으로 구성된 문자열 배열로 출력하라.
매개변수 | 값 |
---|---|
n | 5 |
arr1 | [9, 20, 28, 18, 11] |
arr2 | [30, 1, 21, 17, 28] |
출력 | ["#####","# # #", "### #", "# ##", "#####"] |
매개변수 | 값 |
---|---|
n | 6 |
arr1 | [46, 33, 33 ,22, 31, 50] |
arr2 | [27 ,56, 19, 14, 14, 10] |
출력 | ["######", "### #", "## ##", " #### ", " #####", "### # "] |
class Solution {
public String[] solution(int n, int[] arr1, int[] arr2) {
String[] answer = new String[n];
for(int i=0; i<n; i++) {
String row = "";
String[] a1 = Integer.toBinaryString(arr1[i]).split("");
String[] a2 = Integer.toBinaryString(arr2[i]).split("");
for(int j=0; j<n; j++) {
if("1".equals(a1[j]) || "1".equals(a2[j])) row += "#";
else row += " ";
}
answer[i] = row;
}
return answer;
}
}
당차게 풀었으나 오류가 난 코드
Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: Index 5 out of bounds for length 5
at Solution.solution(Unknown Source)
at SolutionTest.lambdamain$0(Unknown Source) at SolutionTestSolutionRunner.run(Unknown Source)
at SolutionTest.main(Unknown Source)
인덱스가 길이를 벗어난다고 한다.
뭐가 문제야...하고 문제를 겁나 째려보다가 알게된 사실!
이진법으로 표현한 수가 n
보다 길이가 적다면 앞에 0을 추가해줘야한다.
class Solution {
public String[] addZero(String bin, int n) {
if(bin.length() != n) {
String zero = "";
for(int i=0; i<n-bin.length(); i++) {
zero += "0";
}
bin = zero + bin;
}
return bin.split("");
}
public String[] solution(int n, int[] arr1, int[] arr2) {
String[] answer = new String[n];
for(int i=0; i<n; i++) {
String row = "";
String bin1 = Integer.toBinaryString(arr1[i]);
String bin2 = Integer.toBinaryString(arr2[i]);
String[] a1 = addZero(bin1, n);
String[] a2 = addZero(bin2, n);
for(int j=0; j<n; j++) {
if("1".equals(a1[j]) || "1".equals(a2[j])) row += "#";
else row += " ";
}
answer[i] = row;
}
return answer;
}
}
addZero
라는 부족한 만큼 0을 추가해주는 함수를 만들어서 해결했다.
굳이 함수를 만든 이유는 arr1
과 arr2
를 돌려야하므로 코드 중복이 되지 않게하기 위함이다.
완성된 코드를 설명해보자면 n
만큼 반복문을 돌면서
arr1
과 arr2
의 요소를 각각 Integer.toBinaryString()
을 이용해 2진수법으로 저장된 String bin1
, bin2
를 만든다.
그리고 addZero
함수에 넣어 n
보다 길이가 적다면 앞에 부족한 만큼 0을 넣어준 후 split()
를 사용하여 문자열 배열로 만들어 반환한다.
두 문자열 배열에 한 곳이라도 1
이 있다면 지도 한 줄을 나타내는 row
에 #
을 추가하고 1
이 없다면 공란을 추가한다.
그렇게 지도 한 줄이 완성되면 answer
에 추가해준다.
class Solution {
public String[] solution(int n, int[] arr1, int[] arr2) {
String[] answer = new String[n];
String temp;
for(int i = 0 ; i < n ; i++){
temp = String.format("%16s", Integer.toBinaryString(arr1[i] | arr2[i]));
temp = temp.substring(temp.length() - n);
temp = temp.replaceAll("1", "#");
temp = temp.replaceAll("0", " ");
answer[i] = temp;
}
return answer;
}
}
아 비트연산자로 둘 중 하나라도 1이 있으면 1로 받는 방법도 있었다.
다른 사람들은 길이를 "%"+n+"s"
로 받았는데 이사람은 16으로 처음부터 지정한 후에 substring()
으로 자르는 방법을 선택했다. 이게 더 빠르다고 한다.
감자의 코드보다 훨씬 깔끔하고 이쁘다.