[programmers] [1차] 비밀지도 - Java

[1차] 비밀지도

문제 설명

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.

2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.

3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.

4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

입력 형식

• 1 ≦ n ≦ 16

• arr1, arr2는 길이 n인 정수 배열로 주어진다.

• 정수 배열의 각 원소 x를 이진수로 변환했을 때의 길이는 n 이하이다. 즉, 0 ≦ x ≦ 2 - 1을 만족한다.
n

출력 형식

입출력 예제

solution.java

import java.util.*;

class Solution {
    public String[] solution(int n, int[] arr1, int[] arr2) {
        String[] answer = new String[n];
        
        for (int i = 0; i < n; i++) {
            StringBuilder sb = new StringBuilder();
            int merged = arr1[i] | arr2[i]; // OR 연산으로 지도 합치기
            // 비트 연산 결과를 문자열로 변환하여 지도 생성
            for (int j = 0; j < n; j++) {
                if ((merged & 1) == 1) {
                    sb.insert(0, "#"); // 벽인 경우 #
                } else {
                    sb.insert(0, " "); // 공백인 경우 공백 추가
                }
                merged >>= 1; // 오른쪽 시프트하여 다음 비트로 이동
            }
            
            answer[i] = sb.toString();
        }
        
        return answer;
    }
}

핵심 키워드

• OR 연산(arr1[i] | arr2[i])을 통해 각 지도에서 벽의 위치를 구할 수 있다.

• AND 연산(merged & 1)을 통해 제일 오른쪽 비트가 1인지의 여부를 구할 수 있다.

• sb.insert(0, "#")를 통해 문자열을 지정한 인덱스에 삽입할 수 있다.

• 시프트 연산(merged >>= 1)을 통해 비트를 한 칸씩 해당 방향으로 이동시킬 수 있다.

결론!