[프로그래머스/C++ 문제 풀이] Lv. 1 - [1차] 비밀지도

문제 설명

네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

네오가 프로도의 비상금을 손에 넣을 수 있도록, 비밀지도의 암호를 해독하는 작업을 도와줄 프로그램을 작성하라.

제한 사항

입력으로 지도의 한 변 크기 n 과 2개의 정수 배열 arr1, arr2가 들어온다.

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

출력 형식

원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

입출력 예

---

문제 풀이

이번 문제는 arr1과 arr2의 지도에서 겹친 부분을 찾는 문제입니다.

 

지도의 가로줄은 숫자로 주어지며, 해당 숫자를 2진수로 표현해 벽을 체크해야 하며, 숫자를 2로 나눠가며 코드를 작성할 수도 있지만, 비트마스킹을 통해 해당 수의 특정 위치의 비트가 켜져있는지 체크하는 방식으로 문제를 해결하였습니다.

 

> temp[i] & (1 << j) : temp의 i번째 숫자의 j번째 비트가 켜져있으면 true, 꺼져있으면 false를 반환하는 조건입니다.

 

예를 들어 9라는 숫자는 2진수로 1001이며, j가 0과 3인 경우 1과 and 연산을 통해 true를 반환하게 되는 방식입니다.

정답 코드

풀이 시간 : 12m 30s

#include <string>
#include <vector>


using namespace std;

vector<char> adj[18];

vector<string> solution(int n, vector<int> arr1, vector<int> arr2) {
    vector<string> answer;
    
    vector<int> temp;
    
    // 비트 합치기 (arr1과 arr2에서 중복으로 켜진 비트를 저장)
    for (int i = 0; i < n; i++)
    {
        temp.push_back(arr1[i] | arr2[i]);
    }
    
    for (int i = 0; i < n; i++)
    {
        for (int j = n - 1; j >= 0; j--)
        {
            // j번째 비트가 켜져 있다면?
            if (temp[i] & (1 << j))
            {
                adj[i].push_back('#');
            }
            // j번째 비트가 꺼져있다면?
            else
            {
                adj[i].push_back(' ');
            }
        }
    }
    
    // adj에 저장된 값을 토대로 최종 출력 결과물을 만들어줍니다.
    for (int i = 0; i < n; i++)
    {
        string tempStr;
        for (int j = 0; j < n; j++)
        {
            tempStr += adj[i][j];
        }
        
        answer.push_back(tempStr);
    }
    
    return answer;
}