[C++] 백준(BOJ) - 1535 : 안녕

인프런에 있는 큰돌님의 강의 10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트를 듣고 정리한 필기입니다.

문제

BOJ - 1535 : 안녕(링크)

세준이는 성형수술을 한 후에 병원에 너무 오래 입원해 있었다. 이제 세준이가 병원에 입원한 동안 자기를 생각해준 사람들에게 감사하다고 말할 차례이다.

세준이를 생각해준 사람은 총 N명이 있다. 사람의 번호는 1번부터 N번까지 있다. 세준이가 i번 사람에게 인사를 하면 L[i]만큼의 체력을 잃고, J[i]만큼의 기쁨을 얻는다. 세준이는 각각의 사람에게 최대 1번만 말할 수 있다.

세준이의 목표는 주어진 체력내에서 최대한의 기쁨을 느끼는 것이다. 세준이의 체력은 100이고, 기쁨은 0이다. 만약 세준이의 체력이 0이나 음수가 되면, 죽어서 아무런 기쁨을 못 느낀 것이 된다. 세준이가 얻을 수 있는 최대 기쁨을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N(≤ 20)이 들어온다. 둘째 줄에는 각각의 사람에게 인사를 할 때, 잃는 체력이 1번 사람부터 순서대로 들어오고, 셋째 줄에는 각각의 사람에게 인사를 할 때, 얻는 기쁨이 1번 사람부터 순서대로 들어온다. 체력과 기쁨은 100보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 세준이가 얻을 수 있는 최대 기쁨을 출력한다.

알고리즘

  1. DP
  2. 냅색

풀이

  • BOJ - 12865 : 평범한 배낭 문제와 동일
  • n개의 물품을 1개씩 담아야 하는 경우 DP를 오른쪽에서부터 채워 나간다
  • 배열 두개에 담아서 문제 풀이도 가능

코드

// BOJ - 1535 : 안녕
#include<bits/stdc++.h>
using namespace std;

int n, a, dp[101], cost[101], happy[101];

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

    cin >> n;

    for (int i = 0; i < n; i++) cin >> cost[i];
    for (int i = 0; i < n; i++) cin >> happy[i];

    for (int i = 0; i < n; i++)
    {
        for (int j = 100; j > cost[i]; j--)
        {
            dp[j] = max(dp[j], dp[j - cost[i]] + happy[i]);
        }
    }
    cout << dp[100] << '\n';

    return 0;
}

평가

  • 냅색 문제
    • N개의 물품을 무한대로 담을 수 있는 경우 => DP 배열을 왼쪽에서부터 채워나가기
    • N개의 물품을 최대 1개 담을 수 있는 경우 => DP 배열을 오른쪽에서부터 채워나가기

© 2023 Jinsoo Lee. All rights reserved.

Powered by Hydejack v9.1.6