🍞 Problem Solving/Programmers

[프로그래머스][Level1] [1차] 비밀지도 c++

박빵이 2022. 9. 2. 00:48

풀이과정

v1 문자열 벡터에 arr1의 값을 이진수로 변환한 문자열을 넣고,  
v2 문자열 벡터에 arr2의 값을 이진수로 변환한 문자열을 넣는다.

이진수로 변환할 때 9일 경우 1001로 n = 5일 때 5자리를 만족하지 못한다.  
5자를 맞춰주기 위해 앞부분에 0을 붙여주어 01001을 만든다.

그리고 이중 for문을 돌려 만약 두 문자열 중 한 곳이라도 1(벽)이 있다면 #을 출력하고,  
둘 다 0(공백)이라면 공백을 출력하도록 하였다.

 

풀이1

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

using namespace std;

string binaryNum(int n, int idx){
    string str = "";
    while(n != 0){
        str += to_string(n % 2);
        n /= 2;
    }
    while(str.size() != idx) str += "0";
    reverse(str.begin(), str.end());
    return str;
}

vector<string> solution(int n, vector<int> arr1, vector<int> arr2) {
    vector<string> answer;

    vector<string> v1;
    vector<string> v2;
    for(int i = 0; i < n; i++){
        v1.push_back(binaryNum(arr1[i], n));
        v2.push_back(binaryNum(arr2[i], n));
    }
    for(int i = 0; i < n; i++){
        string str = "";
        for(int j = 0; j < n; j++){
            if(v1[i][j] == '1' || v2[i][j] == '1'){
                str += "#";
            }
            else str += " ";
        }
        answer.push_back(str);
    }
    return answer;
}

 

풀이2

reverse를 구현하는 대신 str = 더할 것 + str 식으로 처음부터 문자열 앞에 붙여줬다.

그 대신 이 방식은 느리므로 시간 초과가 날 수 있다.

 

#include <string>
#include <vector>

using namespace std;

string binaryNum(int n, int idx){
    string str = "";
    while(n != 0){
        str = to_string(n % 2) + str;
        n /= 2;
    }
    while(str.size() != idx) str = "0" + str;
    return str;
}

vector<string> solution(int n, vector<int> arr1, vector<int> arr2) {
    vector<string> answer;

    vector<string> v1;
    vector<string> v2;
    for(int i = 0; i < n; i++){
        v1.push_back(binaryNum(arr1[i], n));
        v2.push_back(binaryNum(arr2[i], n));
    }
    for(int i = 0; i < n; i++){
        string str = "";
        for(int j = 0; j < n; j++){
            if(v1[i][j] == '1' || v2[i][j] == '1'){
                str += "#";
            }
            else str += " ";
        }
        answer.push_back(str);
    }
    return answer;
}