[Programmers] 숫자 블록

문제 설명

그렙시에는 0으로 된 도로에 숫자 블록을 설치하기로 하였습니다. 숫자 블록의 규칙은 다음과 같습니다.

블록의 번호가 n 일 때, 가장 처음 블록은 n 2번째 위치에 설치합니다. 그다음은 n 3, 그다음은 n * 4, ...로 진행합니다.만약 기존에 블록이 깔려있는 자리라면 그 블록을빼고 새로운 블록으로 집어넣습니다.

예를 들어 1번 블록은 2,3,4,5, ... 인 위치에 우선 설치합니다. 그다음 2번 블록은 4,6,8,10, ... 인 위치에 설치하고, 3번 블록은 6,9,12... 인 위치에 설치합니다.

이렇게 3번 블록까지 설치하고 나면 첫 10개의 블록은 0, 1, 1, 2, 1, 3, 1, 2, 3, 2이됩니다.

그렙시는 길이가 1,000,000,000인 도로에 1번 블록부터 시작하여 10,000,000번 블록까지 위의 규칙으로 모두 놓았습니다.

그렙시의 시장님은 특정 구간의 어떤 블록이 깔려 있는지 알고 싶습니다.

구간을 나타내는 두 수 begin, end 가 매개변수로 주어 질 때, 그 구간에 깔려 있는 블록의 숫자 배열(리스트)을 return하는 solution 함수를 완성해 주세요.

제한 사항

• begin, end 는 1 이상 1,000,000,000이하의 자연수 이고, begin는 항상 end보다 작습니다.
• end - begin 의 값은 항상 10,000을 넘지 않습니다.

소스 코드

class Solution {
    public int[] solution(long begin, long end) {
        int beginInt = (int) begin;
        int endInt = (int) end;
        int[] answer = new int[endInt-beginInt+1];
        
        int idx = 0;
        for(int i = beginInt; i < endInt+1; i++) {
            answer[idx] = block(i);
            idx += 1;
        }
        
        return answer;
    }
    
    public int block(int n) {
        if(n == 1) return 0;
        
        for(int i = 2; i*i <= n; i++) {
            if(n%i == 0 && n/i <= 10000000) { //블럭은 10000000까지만 가능
                return n/i; //가장 작은 약수로 나눈 값 = 가장 큰 약수
            }
        }
        
        return 1; //소수이면 1
    }
}

Review

에라토스테네스의 체를 이용해 소수를 구하는 법을 활용한 문제
사용 가능한 블럭이 10000000번 까지인 것을 주의해야 한다 🤧