[c++] 프로그래머스 - [1차] 비밀 지도(Level 1)

문제

코딩테스트 연습 - [1차] 비밀지도

 

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

char map[17][17];

vector<string> solution(int n, vector<int> arr1, vector<int> arr2) {
    vector<string> answer;
    for(int i=0; i<arr1.size(); i++){
        int cnt = n;
        int num = arr1[i];
        string bit = "";
        while(cnt--){
            bit+=(num%2)+'0';
            num/=2;
        }
        reverse(bit.begin(),bit.end());
        for(int k=0; k<n; k++){
            if(bit[k]=='0'&&map[i][k]!='#') map[i][k] = ' ';
            else if(bit[k]=='1') map[i][k] = '#';
        }
    }
    for(int i=0; i<arr2.size(); i++){
        int cnt = n;
        int num = arr2[i];
        string bit = "";
        while(cnt--){
            bit+=(num%2)+'0';
            num/=2;
        }
        reverse(bit.begin(),bit.end());
        for(int k=0; k<n; k++){
            if(bit[k]=='0'&&map[i][k]!='#') map[i][k] = ' ';
            else if(bit[k]=='1') map[i][k] = '#';
        }
    }
    
    for(int i=0; i<n; i++){
        string inp ="";
        for(int k=0; k<n; k++){
            inp+=map[i][k];
        }
        answer.push_back(inp);
    }
    return answer;
}

 

풀이(20분)

이진법을 구한 후에 각 이진수의 자릿수가 1이면 '#' 0이면 ' '처리를 해주어 가면서 구현했습니다. 다만 이 문제는 이진수의 길이가 n으로 지정되어서 주어지기 때문에 이진수를 구할 때 n==0이 될 때까지 구하는 게 아닌 n만큼 나누어가면서 구해야 합니다.