[백준 2156번] 포도주 시식

풀이

---

계단 오르기문제랑 비슷한 문제이다.

연속으로 세잔을 마시지 못하는 조건이 들어간것도 똑같다.

모든 경우의 수를 구해서 답을 구할 수 있지만 시간초과가 나므로 DP로 해결할 수 있다.

마실 수 있는 경우의 수를 3가지로 분류할 수 있다.

1. 현재 잔을 마셧을 경우

2. 이전잔과 현재잔 전전전잔을 마셧을 경우

3. 전전잔과 현재잔을 마셧을 경우

이 경우의 수의 최대값을 계속 DP배열에 저장해나가면 해결할 수 있다.

1. dp[i-1]
2. wine[i] + wine[i - 1] + dp[i - 3]
3. wine[i] + dp[i - 2];

소스 코드

---

#include <iostream>
using namespace std;
 
int n, wine[10001],dp[10001];
 
int Max(int a, int b) {
    return a > b ? a : b;
}
 
int main() {
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> wine[i];
    }
 
    dp[0] = wine[0];
    dp[1] = wine[0] + wine[1];
    if (n == 1 || n == 2) {
        cout << dp[n - 1] << '\n';
        return 0;
    }
    for (int i = 2; i <= n; i++) {
        dp[i] = Max(wine[i] + wine[i - 1] + dp[i - 3],Max(dp[i-1], wine[i] + dp[i - 2]));
    }
    cout << dp[n] << '\n';
    return 0;
}