[프로그래머스] 숫자 블록

1 minute read

문제 설명

그렙시에는 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을 넘지 않습니다.

풀이

60분
주어진 구간의 숫자들의 약수 중 최댓값을 구하는 문제이다. 처음에는 별 생각없이 풀었다가 효율성 테스트를 통과하지 못하고 다시 문제를 풀었다. 구하려는 값의 제곱근까지만 반복문을 돌려 구하는 방식으로 바꾸었는데, 이번에는 효율성 테스트에서 실패하는 문제가 발생했다.

1번 블록부터 시작하여 10,000,000번 블록까지

의 조건을 추가해주어서 통과할 수 있었다. 레벨4 문제들 중 매우 쉬운편 이었는데 빠르게 풀어내지 못해 아쉽다.

시간 초과 코드

// 11:12 ~ 
#include <string>
#include <vector>

using namespace std;

vector<int> solution(long long begin, long long end) {
    vector<int> answer;
    answer.resize(end - begin + 1, 0);
    for(int i = begin; i <= end; ++i){
        for(int j = i / 2; j >= 1; --j){
            if(i % j == 0){
                answer[i - begin] = j;
                break;
            }
        }
    }
    return answer;
}

코드

// 11:12 ~ 12:12

#include <string>
#include <vector>


using namespace std;

vector<int> solution(long long begin, long long end) {
    vector<int> answer;
    answer.resize(end - begin + 1, 1);
    
    if(begin == 1) answer[0] = 0;
    for(int i = begin; i <= end; ++i){
        for(int j = 2; j * j <= i; ++j){
            if(i % j == 0 && i / j <= 10000000){
                answer[i - begin] = i / j;
                break;
            }
        }
    }
    return answer;
}