프로그래머스 - n진수 게임(Swift)

프로그래머스 - n진수 게임(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 

 나의 풀이

 

 

문제를 풀 때 "현재의 turn을 사람의 숫자로 나눠서 떨어지는 나머지가 순서와 같다면 결과값에 넣는다"는 구조를 파악하는데는 오래걸리지 않았지만, 사람의 순서와 사람 수가 같을 때는 나머지가 0인 경우지만 값을 넣어줘야하기 때문에 else if로 빼는 로직을 생각하는데는 조금 오래걸렸다. 

 

그리고 이 코드는 마지막 줄에 return ""가 있기 때문에 왠지 모르게 좋지 않은 코드라는 느낌이 온다.

그래서 다른 사람의 코드를 살펴봤다.

 

func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {
    var turn = 1
    var ans = ""
    let arr = (0..<t*m).map{ String($0, radix: n).uppercased() }
    for x in arr {
        for j in x { 
            if turn % m == p {
                ans.append(j)
                if ans.count == t {
                    return ans
                }
            } else if p == m && turn % p == 0 {
                ans.append(j)
                if ans.count == t {
                    return ans
                }
            }
            turn += 1
        }    
    }
    return ""
}

 

 다른 사람의 풀이

 

내 코드와 비교해 봤을 때 훨씬 효율적이고 직관적이다.

 

나는 쌓여가는 answer의 카운트가 m*t인 count보다 작을때 까지 while문을 순회하는 방법이 아닌, answer.count == t가 되었을 때 리턴을 했다. 즉 해당하는 값만을 answer에 넣었고 아래 풀이의 사람은 먼저 인원수만큼 answer에 추가를 해준 뒤 stride를 통해 순서에 해당하는 값만을 추출하고, reduce해서 풀이를 했다.

 

func solution(_ n:Int, _ t:Int, _ m:Int, _ p:Int) -> String {
    var ans:[String] = []
    let count = m*t
    var number = 0
    while ans.count < count {
        ans += String(number, radix: n).map{ $0.uppercased() }
        number += 1
    }
    
    return stride(from: p-1, to: count, by: m).reduce("", {$0 + ans[$1]})
}

 

 피드백

 

stride는 배열처럼 사용할 수 있다. (단 map이나 reduce처럼 형 변환을 확실하게 하고 사용해야 한다.)