관리 메뉴

Storage Gonie

(5) [C++, Java] 백준 No.11052 : 카드 구매하기 본문

알고리즘/백준풀이6. 다이나믹프로그래밍

(5) [C++, Java] 백준 No.11052 : 카드 구매하기

Storage Gonie 2019. 4. 30. 22:27
반응형

문제

풀이

- 자세한 설명 https://ldgeao99.tistory.com/entry/챕터3-6-다이나믹-프로그래밍-문제-풀이5-백준-No11052-붕어빵-판매하기-문제

# C++(Top-down 방식)
- Top-down방식은 전역변수를 사용해야 구현이 쉽다.
- Bottom-up방식에서 2중 for문 중 바깥 for문을 제거하고 j를 i로, i를 n으로 수정했다고 생각하면됨.
- 재귀적인 호출이 일어날 때는 두번째 if(d[n] > 0) return d[n]; 을 꼭 넣어줘야 한다.
(다이나믹 프로그래밍의 필수적요소)

#include <iostream>

using namespace std;

int d[1001] = {0};
int p[100001] = {0};

int getMaxPrice(int n)
{
    if(n == 0)
        return 0;

    else if (d[n] > 0)           // 이게 없으면 시간 초과됨.
        return d[n];

    for(int i = 1; i <= n; i++)  // 선택가능한 것들 중 max 가격을 선택하기 위한 내부 for문
        d[n] = max(d[n], getMaxPrice(n-i) + p[i]);

    return d[n];
}

int main()
{
    int T;
    cin >> T;
    
    // 묵음별 가격을 입력받음
    for (int i = 1; i <= T; i++)
        cin >> p[i];
    
    cout << getMaxPrice(T);
}


# C++(Bottom-up 방식)
- 배열 대신 메모리를 효과적으로 관리하기위해 벡터를 사용할 수도 있음

#include <iostream>

using namespace std;

int d[1001] = {0};
int p[100001] = {0};

int getMaxPrice(int n)
{
    for (int i = 1; i <= n; i++){
        for(int j = 1; j <= i; j++)  // 선택가능한 것들 중 max 가격을 선택하기 위한 내부 for문
            d[i] = max(d[i], d[i-j] + p[j]);
    }
    
    return d[n];
}

int main()
{
    int T;
    cin >> T;
    
    // 묵음별 가격을 입력받음
    for (int i = 1; i <= T; i++)
        cin >> p[i];
    
    cout << getMaxPrice(T);
}
#include <iostream>
#include <vector>   // vector를 사용하기 위함

using namespace std;

int main()
{
    int T;
    cin >> T;

    // 묵음별 가격을 입력받음
    vector<int> p_vec(T+1);       // 묶음별 가격, p_vec[0]은 사용안하므로 T+1 만큼 크기를 줘서 선언함.
    for (int i = 1; i <= T; i++)
        cin >> p_vec[i];
    
    // d를 전역변수가 아니라 아래와 같이 벡터로 선언하게 되면 재귀적인 함수를 호출하는 Top-down 방식을 쓰기 어려우므로 Bottom-up 방식을 사용하자.
    vector<int> d_vec(T+1);
    for(int i = 1; i <= T; i++){      // d_vec[1]부터 시작해서 d_vec[T]까지 차례로 구해나가기 위한 바깥 for문
        for(int j = 1; j <= i; j++){  // 선택가능한 것들 중 max 가격을 선택하기 위한 내부 for문
            d_vec[i] = max(d_vec[i], d_vec[i-j] + p_vec[j]);
        }
    }

    cout << d_vec[T];
}

# Java

import java.util.*;

public class Main {
    public static void main(String args[]) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n+1];
        for (int i=1; i<=n; i++) {
            a[i] = sc.nextInt();
        }
        int[] d = new int[n+1];
        for (int i=1; i<=n; i++) {
            for (int j=1; j<=i; j++) {
                if (d[i] < d[i-j] + a[j]) {
                    d[i] = d[i-j] + a[j];
                }
            }
        }
        System.out.println(d[n]);
    }
}

 

반응형
Comments