문제
세준이는 성형수술을 한 후에 병원에 너무 오래 입원해 있었다. 이제 세준이가 병원에 입원한 동안 자기를 생각해준 사람들에게 감사하다고 말할 차례이다.
세준이를 생각해준 사람은 총 N명이 있다. 사람의 번호는 1번부터 N번까지 있다. 세준이가 i번 사람에게 인사를 하면 L[i]만큼의 체력을 잃고, J[i]만큼의 기쁨을 얻는다. 세준이는 각각의 사람에게 최대 1번만 말할 수 있다.
세준이의 목표는 주어진 체력내에서 최대한의 기쁨을 느끼는 것이다. 세준이의 체력은 100이고, 기쁨은 0이다. 만약 세준이의 체력이 0이나 음수가 되면, 죽어서 아무런 기쁨을 못 느낀 것이 된다. 세준이가 얻을 수 있는 최대 기쁨을 출력하는 프로그램을 작성하시오.
평범한 배낭 문제와 같은 방식으로 접근했다.
현재 재귀함수를 통해 hp를 전달하며 현재 사람에게 감사할지 말지를 정한다.
//index N일때 최대 earnGlad값
int EarnGlad(int index,int N, int curHp)
{
if (index == N) return 0;
int& ret = dp[index][curHp];
if (ret != -1) return ret;
//우선 n포함 안했을 때의 값
ret = EarnGlad(index + 1, N,curHp);
//n번째 포함했을 때의 값
if(curHp - minusHpArray[index]>0)
ret = max(ret, EarnGlad(index + 1, N, curHp - minusHpArray[index]) + earnGladArr[index]);
return ret;
}
#include <string>
#include <vector>
#include<queue>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int dp[21][101], minusHpArray[21], earnGladArr[21];
//index N일때 최대 earnGlad값
int EarnGlad(int index,int N, int curHp)
{
if (index == N) return 0;
int& ret = dp[index][curHp];
if (ret != -1) return ret;
//우선 n포함 안했을 때의 값
ret = EarnGlad(index + 1, N,curHp);
//n번째 포함했을 때의 값
if(curHp - minusHpArray[index]>0)
ret = max(ret, EarnGlad(index + 1, N, curHp - minusHpArray[index]) + earnGladArr[index]);
return ret;
}
int main()
{
int N;
fill(&dp[0][0], &dp[21][101], -1);
cin >> N;
for(int i=0;i<N;i++)
cin >> minusHpArray[i];
for (int i = 0; i < N; i++)
cin >> earnGladArr[i];
cout<<EarnGlad(0, N, 100);
}
처음에 dp[index]로만 접근했는데, 해당 index에 접근했을 때 어떤 hp여야 가장 큰 답이 나오는지도 고려해야한다.
따라서 dp[index][hp] 이런 식으로 이차원 배열을 사용해야한다.