[Swift]프로그래머스_n진수 게임

문제 설명

튜브가 활동하는 코딩 동아리에서는 전통적으로 해오는 게임이 있다. 이 게임은 여러 사람이 둥글게 앉아서 숫자를 하나씩 차례대로 말하는 게임인데, 규칙은 다음과 같다.

1. 숫자를 0부터 시작해서 차례대로 말한다. 첫 번째 사람은 0, 두 번째 사람은 1, … 열 번째 사람은 9를 말한다.
2. 10 이상의 숫자부터는 한 자리씩 끊어서 말한다. 즉 열한 번째 사람은 10의 첫 자리인 1, 열두 번째 사람은 둘째 자리인 0을 말한다.

이렇게 게임을 진행할 경우,
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 0, 1, 1, 1, 2, 1, 3, 1, 4, …
순으로 숫자를 말하면 된다.

 

한편 코딩 동아리 일원들은 컴퓨터를 다루는 사람답게 이진수로 이 게임을 진행하기도 하는데, 이 경우에는
0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 1, 1, 1, …
순으로 숫자를 말하면 된다.

이진수로 진행하는 게임에 익숙해져 질려가던 사람들은 좀 더 난이도를 높이기 위해 이진법에서 십육진법까지 모든 진법으로 게임을 진행해보기로 했다. 숫자 게임이 익숙하지 않은 튜브는 게임에 져서 벌칙을 받는 굴욕을 피하기 위해, 자신이 말해야 하는 숫자를 스마트폰에 미리 출력해주는 프로그램을 만들려고 한다. 튜브의 프로그램을 구현하라.

 

튜브가 말해야 하는 숫자 t개를 공백 없이 차례대로 나타낸 문자열. 단, 10~15는 각각 대문자 A~F로 출력한다.

 

진법 n, 미리 구할 숫자의 갯수 t, 게임에 참가하는 인원 m, 튜브의 순서 p 가 주어진다.

 

예시

    n      t      m    p           result

2	4	2	1	"0111"
16	16	2	1	"02468ACE11111111"
16	16	2	2	"13579BDF01234567"

 

풀이

처음에 내가 푼 방법은 세 가지 테스트케이스에서 시간초과가 나왔다. 

먼저, while문으로 1씩 증가해가며 n진수로 바꾼 수를 한자리 씩 문자열에 추가해주었다. 그러다 문자열의 길이가 t보다 크거나 같아졌을 때, 반복문을 탈출하게 하였다.

다음으로 만들어진 문자열을 하나씩 확인해가며 본인의 순서가 왔을때의 문자를 따로 결과 문자열에 저장하여 리턴하도록 하였다. 이때, 순서를 계산하기 위해 따로 변수를 선언하지 않고 나머지를 이용에 순서를 계산하였다. enumerated를 이용해 구한 인덱스로 나머지를 계산하였다.

import Foundation

func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {

    var numStr: String = ""
    var result: String = ""
    var count: Int = 0
    var checkCount: Int = 0

    // 일단 진수에 맞는 총 문자열을 구한 뒤에 membercount를 늘려가며 순서가 될 때만 그 문자열을 result에 넣어서 출력
    while true {
        numStr.append(String(count, radix: n))
        if numStr.count >= t * m {
            break
        } else {
            count += 1
        }
    }
    // 문자열을 대문자로 바꿔줌
    numStr = numStr.uppercased()
    // 나머지를 이용해서 순서를 계산하기 위해서는 p가 마지막인 경우는 따로 생각해줘야함
    if p == m {
        for (i, num) in numStr.enumerated() {
            if (i + 1) % m == 0 { // 나머지를 이용해 순서계산
                result.append(String(num)) // +에서 append로 바꿔봄
                checkCount += 1
            }
            if checkCount == t {
                break
            }
        }
    } else {
        for (i, num) in numStr.enumerated() {

            if (i + 1) % m == p { // 나머지를 이용해 순서계산
                result.append(String(num))
                checkCount += 1
            }
            if checkCount == t {
                break
            }
        }
    }
    return result
}

위처럼 해결하니 시간초과가 나왔다. 그래서 다른 분의 답을 참고하여 필요없는 연산을 줄여보았다. 나머지를 이용해 계산하던 순서도 그냥 변수를 이용해 해결하기로 하였고, 미리 문자열을 전부 구해놓은 뒤에 다시 반복문을 돌아가며 넣지 않고 숫자 하나씩 그때그때 확인하기로 하였다. 

앞에서 구한 코드에서는 문자열을 구할 때 t*m 크기의 문자열을 구하였기에 시간초과가 났던 것 같다.

 

import Foundation

func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {
    
    var result: String = ""
    var count: Int = 0
    var order: Int = 1
    
    while result.count < t {
        let numStr = String(count, radix: n, uppercase: true)
        for i in numStr {
            if order == p {
                result.append(String(i))
                if result.count == t {
                    break
                }
            }
            order += 1
            if order > m {
                order = 1
            }
        }
        count += 1
    }
    return result
}

새로 작성한 코드에서 다른 점은 order라는 순서를 나타내는 변수를 이용해 나누기 연산을 줄였다. 또한 미리 문자열을 구한 뒤 다시 반복문을 돌리는 것이 아닌 하나씩 확인해가기때문에 필요한 횟수만 반복할 수 있게 되었다.

 

결론

오랜만에 푸는 문젠데 잠깐 안풀었다고 해결하는데 굉장히 오래 걸렸다. 앞으로는 꾸준히 풀어야 감을 잊지 않을 것 같다. 조만간 3단계도 풀어봐야겠다.