n진수 게임

문제

여러 사람이 둥글게 앉아 숫자를 0부터 차례대로 말하는 게임을 진행합니다. 10 이상의 숫자는 한 자리씩 끊어서 말해야 하며, 이진법에서 십육진법까지 다양한 진법으로 게임을 진행할 수 있습니다. 게임에 참가하는 인원과 튜브의 순서가 주어질 때, 튜브가 미리 준비해야 할 숫자들을 차례대로 문자열로 구하는 프로그램을 작성하는 문제입니다.

• 입력:
  • 진법 n (2 ≤ n ≤ 16)
  • 미리 구할 숫자의 개수 t (0 < t ≤ 1000)
  • 게임에 참가하는 인원 m (2 ≤ m ≤ 100)
  • 튜브의 순서 p (1 ≤ p ≤ m)
• 출력: 튜브가 말해야 하는 숫자 t개를 공백 없이 차례대로 나타낸 문자열 (10~15는 대문자 A~F로 출력)

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

풀이: 진법 변환과 문자열 인덱싱을 활용한 구현

자바의 내장 함수인 Integer.toString(num, radix)를 사용하면 진법 변환을 매우 쉽게 처리할 수 있습니다. 전체 게임의 흐름을 하나의 긴 문자열로 만들고, 튜브의 차례에 해당하는 인덱스만 솎아내는 직관적인 방법입니다.

• 전체 문자열 생성 및 최적화: 0부터 m * t - 1까지의 숫자를 1씩 증가시키며 n진수로 변환해 StringBuilder에 이어 붙입니다. 튜브가 t개의 문자를 찾기 위한 게임의 총 턴 수는 최대 m * t입니다. 각 숫자는 최소 1자리 이상을 차지하므로, 숫자를 m * t번 이어 붙이면 전체 문자열의 길이는 무조건 m * t를 넘게 되어 인덱스 초과 오류를 방지할 수 있습니다.
• 튜브의 순서 추출: 전체 게임 문자열이 만들어지면, 튜브의 첫 번째 순서 인덱스인 p - 1부터 시작합니다. 이후 참가 인원인 m만큼 인덱스를 건너뛰며(idx += m) 문자를 추출해 정답 StringBuilder에 붙입니다.
• 대문자 변환: 10~15에 해당하는 문자는 문제의 조건에 따라 A~F로 출력되어야 하므로, 완성된 정답 문자열을 반환할 때 toUpperCase()를 호출하여 일괄 변환해 줍니다.

코드

class Solution {
    public String solution(int n, int t, int m, int p) {
        StringBuilder sb = new StringBuilder();
        
        // 튜브가 t번 말하기 위해 필요한 최소 보장 횟수만큼 숫자 변환
        for (int i = 0; i < m * t; i++) 
            sb.append(Integer.toString(i, n));
        
        StringBuilder ans = new StringBuilder();
        int idx = p - 1;
        
        // 튜브가 말해야 할 t개의 문자 추출
        while (ans.length() != t) {
            ans.append(sb.charAt(idx));
            idx += m;
        }
        
        return ans.toString().toUpperCase();
    }
}

알고리즘 분류

구현 (Implementation), 문자열 (String), 수학 (Mathematics)