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

안녕하세요 윤교님 ㅎㅎ

다음과 같은 방법으로 시간을 0에서 부터 시작하여 K시간까지 하는 방법으로 코드를 작성하였는데 왜 틀린 답이라고 나올까요?? 만약 저와 같은 방법으로 하려면 어떤 식으로 코드를 수정해야 될까요?

>> 좋은 코드입니다. ㅎㅎ 다만 초기화부분을 -1로 해주셔야 해요. 답의 범위가 0부터 시작되기 때문에 그 범위를 벗어나는 값으로 초기화를 해야 합니다.

제가 윤교님 코드를 좀 다듬어봤는데요. ㅎㅎ 이렇게 해주시면 됩니다.

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

int dp[101][100001];

int n, k; // 구간수, 시간제한
vector<pair<int, int>> walk;
vector<pair<int, int>> bicycle;

int wtime, wmoney, btime, bmoney;

int go(int depth, int n_time) {
    if (n_time > k) return -2000000000;
    if (depth == n) return 0;
    int& ret = dp[depth][n_time];
    if (ret != -1) return ret;

    ret = max(go(depth + 1, n_time + walk[depth].first) + walk[depth].second,
              go(depth + 1, n_time + bicycle[depth].first) + bicycle[depth].second);

    return ret;
}

int main() {
    ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

    cin >> n >> k;
 
    for (int i = 0; i < n; i++) {
        cin >> wtime >> wmoney >> btime >> bmoney;
        walk.push_back({wtime, wmoney});
        bicycle.push_back({btime, bmoney});
    }

    memset(dp, -1, sizeof(dp));

    cout << go(0, 0) << "\n";
    
    return 0;
}

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

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

감사합니다.

강사 큰돌 올림.