[Programmers] n진수 게임

문제 👉 https://programmers.co.kr/learn/courses/30/lessons/17687

 

문제 설명

튜브가 N진수 게임을 하는데 튜브 순서에 말해야 하는 숫자들을 구하는 문제입니다.

게임 규칙은 문제에 자세히 나와있습니다. 

문제 풀이

모든 숫자 결과값을 가진 string 배열을 구하고 그 중에서 튜브의 차례에 말할 숫자들을 뽑아야 합니다. 

이 문제를 풀기 위해서 두가지를 알아야 합니다!!

 

1. N진수 변환 방법

2. for문 돌기 → +1이 아니라 +n만큼 돌아야 합니다. (튜브의 순서는 +사람수만큼 돌아오기 때문)

 

1번은 swift의 radix 메소드를 사용하면 됩니다. (항상 구현했었는데 오늘 처음 암 ㅎㅎ,,)

변환할 수와 진 수를 정하면 됩니다.

uppercase는 optional 파라미터인데 11진수이상부터 알파벳으로 표현되는 숫자들의 대소문자를 결정합니다. 

let target: Int = 3
let n: Int = 2

let result = String(target, radix: n, uppercase: true)
print(result)

// 11

 

2번 방법은 간단하게 생각하면 for문을 돌며 나머지 연산을 이용해서 구할 수 있습니다. 

n, n+2, n+4, n+6 일 경우

2로 나눈 나머지로 판단하여 숫자들을 뽑을 수 있습니다.

// 문제의 풀이 아님!!

for (i,num) in totalNumber.enumerated() {
    if i % 2 == 1 {
        result.append(num)
        if result.count == t {
            break
        }
    }
}

 저는 이 풀이방법을 사용했지만 다른 사람의 풀이를 보고 stride라는 메소드를 알게 되었습니다!!

stride

stride(from: A, to: B, by: C) 라고 하면 

A부터 B까지 C만큼 더하면서 갈거야! 라는 뜻입니다. 

이 때 B는 포함하지 않고 B-1 까지 포함합니다.

for i in stride(from: 0, to: 10, by: 2) {
    print(i)
}

// 0 
// 2
// 4
// 6
// 8

코드

나머지 연산을 이용한 for문

func changeRadix(_ n: Int, _ target: Int) -> String {
    return String(target, radix: n, uppercase: true)
}
func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {
    let total: Int = t * m
    var totalNumber: String = ""
    var result: String = ""
    for i in 0..<total {
        totalNumber.append(changeRadix(n, i))
    }
    for (i,num) in totalNumber.enumerated() {
        if i % m == (p-1) {
            result.append(num)
            if result.count == t {
                break
            }
        }
    }
    
    return result
}

stride 연산을 이용한 for문

func changeRadix(_ n: Int, _ target: Int) -> String {
    return String(target, radix: n, uppercase: true)
}
func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {
    let total: Int = t * m
    var totalNumber: String = ""
    var result: String = ""
    for i in 0..<total {
        totalNumber.append(changeRadix(n, i))
    }
    for i in stride(from: (p-1), to: (p-1) + m * t, by: m) {
        let index = totalNumber.index(totalNumber.startIndex, offsetBy: i)
        result.append(totalNumber[index])
        print(i)
    }
    
    return result
}