인프런 커뮤니티 질문&답변

이종현님의 프로필 이미지
이종현

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

7-F

7-F 1로 만들기 2 질문

해결된 질문

작성

·

328

0

맞은 코드:

http://boj.kr/ed84ee795122447db8c315de4b963c10

틀린 코드 (시간 초과) :

http://boj.kr/fb273c695c8844afb5f3d72ff539970e

안녕하세요 강사님 :)

해당 문제에서 시간 초과가 계속 나서 이것저것 고쳐보다가 결국 정답을 받긴 했는데.. 정답 코드와 틀린 코드가 물론 다른 코드이지만, 맞고 틀리고에 어떤 영향을 준 것인지 이해가 잘 되지 않아서 질문드립니다.

두 코드의 차이점은 간단하게 tracing을 위한 코드인 while문에서의 차이입니다.

맞은 코드는 while문을 첫 번째 정답인 dp[n]의 최솟값만큼 반복합니다.

    int temp = n;
    int repeat = dp[n];
    while(repeat--) {
        cout << temp << ' ';
        temp = prev_[temp];
    }
    cout << 1 << '\n';

틀린 코드는 while문(tracing)을 1이 나올 때까지 반복합니다.

    int temp = n;
    while(1) {
        cout << temp << ' ';
        temp = prev_[temp];
        if(temp == 1) break;
    }
    cout << 1 << '\n';

틀린 코드는 98~99%쯤 시간 초과가 나는데 혹시 이유가 있을까요?

답변 1

1

큰돌님의 프로필 이미지
큰돌
지식공유자

안녕하세요 종현님 ㅎㅎ

살짝 실수하셨는데요.

코드를 보시면 trace를 해가다가 어차피 마지막 지점인 1인 지점까지 출력을 하고 마친다라는 것 아닌가요?

 

수정된 코드는 다음과 같습니다.

#include <bits/stdc++.h>
using namespace std;

int n, dp[1000004], prev_[1000004];

int main()
{
    cin >> n;
    fill(dp, dp + 1000004, 1e9);
    dp[1] = 0;
    for(int i=2; i<=n; i++){
        if(i-1 >= 1){
            dp[i] = dp[i-1] + 1;
            prev_[i] = i-1;
        }
        if(i % 2 == 0 && dp[i] > dp[i/2]+1){
            dp[i] = dp[i/2] + 1;
            prev_[i] = i / 2;
        }
        if(i % 3 == 0 && dp[i] > dp[i/3] + 1){
            dp[i] = dp[i/3] + 1;
            prev_[i] = i / 3;
        }
    }
    cout << dp[n] << '\n';
    int temp = n;
    while(1) {
        cout << temp << ' ';
        if(temp == 1) break;
        temp = prev_[temp]; 
    } 
    return 0;
}

이걸로 제출해보시겠어요?

 

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제가 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

이종현님의 프로필 이미지
이종현

작성한 질문수

질문하기