[programmers] N진수 게임 - Java

N진수 게임

문제 설명

1. 숫자를 0부터 시작해서 차례대로 말한다. 첫 번째 사람은 0, 두 번째 사람은 1, … 열 번째 사람은 9를 말한다.

2. 10 이상의 숫자부터는 한 자리씩 끊어서 말한다. 즉 열한 번째 사람은 10의 첫 자리인 1, 열두 번째 사람은 둘째 자리인 0을 말한다.

입력 형식

• 2 ≦ n ≦ 16

• 0 ＜ t ≦ 1000

• 2 ≦ m ≦ 100

• 1 ≦ p ≦ m

출력 형식

입출력 예제

처음 시도한 코드

class Solution {
    public String solution(int n, int t, int m, int p) {
        String answer = "";
        int cur = 1;
        int num = 0;
        
        while(answer.length()<t){
            String str = Integer.toString(num, n).toUpperCase();
            for(char c: str.toCharArray()){
                if(answer.length()>=t){
                    return answer;
                }
                if(cur%m==p){
                    answer+=c;
                }
                cur++;
            }
            num++;
        }
        
        return answer;
    }
}

개선한 코드

class Solution {
    public String solution(int n, int t, int m, int p) {
        StringBuilder answer = new StringBuilder();
        int cur = 1;
        int num = 0;
        
        while(answer.length() < t){
            String str = Integer.toString(num, n).toUpperCase();
            for(char c: str.toCharArray()){
                if(answer.length() >= t){
                    return answer.toString();
                }
                if(cur % m == p || (cur % m == 0 && p == m)) {
                    answer.append(c);
                }
                cur++;
            }
            num++;
        }
        
        return answer.toString();
    }
}

핵심 키워드

• 처음 시도한 코드는 String에 계속해서 합연산을 사용해서 성능이 좋지 못했고, 해당하는 차례인지 검사하는 조건이 올바르지 않아 몇몇 테스트 케이스를 통과하지 못했다.

• 이를 개선해서 p가 마지막 사람일 때 올바른 출력을 보장하기 위해 p가 m과 같은 경우를 처리하는 조건을 추가하고, StringBuilder를 사용해서 코드를 최적화했다.

결론!