[백준/Baekjoon] 1535 안녕 C++/Python :: Dynamic Programming

https://www.acmicpc.net/problem/1535

1535번: 안녕

---

1. Logic

경우의 수를 나눠보게 되면 인사를 할때와 안할 때 두가지의 경우의 수로 분류할 수 있다. 각 인사를 할지 말지에 대한 고민을 풀이해야 하기 때문에 배낭 문제(KnapSack Problem)라고 볼 수 있다.

---

2. Code

C++

#include<bits/stdc++.h>

using namespace std;

int n;

int dp[21][101];

int lose[20];
int delight[20];

int solve(int idx, int health) {
    if(idx == n) return 0;
    
    int &ret = dp[idx][health];
    if(ret != -1) return ret;
    
    if(health+lose[idx] < 100) {
        ret = max(ret, solve(idx+1, health+lose[idx])+delight[idx]);
    }
    
    ret = max(ret, solve(idx+1, health));
    
    return ret;
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    memset(dp, -1, sizeof(dp));
    
    cin >> n;
    
    for(int i = 0; i < n; i++) {
        cin >> lose[i];
    }
    
    for(int i = 0; i < n; i++) {
        cin >> delight[i];
    }
    
    cout << solve(0, 0);
}

Python

import copy

n = int(input())

dp = [[-1] * 101 for _ in range(21)]

lose = [*map(int, input().split())]
delight = list(map(int,input().split()))

def solve(idx, health):
    if(idx == n): 
        return 0
    
    #ret = dp[idx][health]
    
    if dp[idx][health] != -1:
        return dp[idx][health]
    
    if health+lose[idx] < 100:
        dp[idx][health] = max(dp[idx][health], solve(idx+1, health+lose[idx])+delight[idx])
    
    dp[idx][health] = max(dp[idx][health], solve(idx+1, health))
    
    return dp[idx][health]
    
print(solve(0, 0))

알고리즘 Solve 후 제가 생각한 Logic을 기록하는 개인 공부 블로그입니다.
내용 중 최적화가 가능한 부분등은 언제든지 댓글로 틀린 부분 및 피드백 주시면 공부 및 반영하겠습니다🧐