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

1. 문제 링크 

프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/17681

 

2. 코드

func solution(_ n:Int, _ arr1:[Int], _ arr2:[Int]) -> [String] {
    var answer: [String] = []
    var count : Int = 0
    var str : String = ""
    
    func makeMap(arr: [Int]) -> [Int]{
        var mapArr : [Int] = []
        for i in arr{
            var tempArr : [Int] = []    // 하나씩 결과값을 넣기 위해 임시로 만든 배열
            var share : Int = i         // 몫
            var remain : Int = 0        // 나머지
            
            while share != 0{           //2로 계속 나눌꺼여서 마지막엔 0이된다.
                remain = share % 2      // 2로 나누면 0,1로 나머지가 나옴 
                share = share / 2       // 계속 나눠야 해서
                tempArr.append(remain)   // remain 결과값을 배열에 넣어줌
            }
            
            if tempArr.count < n {          // 1,2(2),4(3),8(4),16(5),64(6)
                                            // 배열안에 갯수 세서 부족한 부분 0으로 채우기
                for _ in 0..<n - tempArr.count{     // 부족한 갯수만큼 반복 배열 뒤에다가 0채우기
                    tempArr.append(0)
                }
            }
            
            tempArr = tempArr.reversed()        // 배열 값을 반전 시키기

            mapArr.append(contentsOf: tempArr)
            
        }
        return mapArr
    }
    // 메서드 호출해서 반환 받은 배열을 같이 zip() 이용해서 돌려서 확인
    for (arr1,arr2) in zip(makeMap(arr: arr1), makeMap(arr: arr2)){
        count += 1          // 카운팅
        if arr1 == 1 || arr2 == 1 {     // 둘중 하나라도 1이라면 벽
            str += "#"
        }else{              // 아닐시는 둘다 0 이므로  빈칸
            str += " "
        }
        if count == n {         // n개씩 잘라야해서
            answer.append(str)      // 정답 배열에 넣고
            count = 0       // 카운팅 0으로 설정
            str = ""        // 문자열 초기화
        }
    }
    return answer
}

3.  코드 설명

makeMap 메서드를 사용하여 입력으로 주어진 arr1과 arr2를 이진수로 변환하여 배열로 만들었습니다.

이진수를 만들기 위해서는 각 숫자를 계속해서 2로 나누고, 나머지를 배열에 저장하여 이진수를 구성합니다. 이진수의 길이가 n보다 작을 때는 0으로 채워주고, 배열을 반전시켜 결과 배열에 추가하였습니다.

두 배열을 동시에 순회하기 위해 zip 함수를 사용하여 함께 반복문을 실행했습니다.

만약 arr1 또는 arr2의 값이 1이라면 벽을 의미하므로 "#"을 추가해줍니다.

매 순회마다 문자열을 str에 저장하는 이유는 각 숫자마다 길이가 n인 문자열을 생성해야 하기 때문입니다.

count를 사용하여 문자열의 길이를 확인하고, 결과 배열에 추가한 후 초기화합니다.