백준 2156 포도주시식

문제

문제링크: https://www.acmicpc.net/problem/2156

풀이전략

1. 문제에 함정이 숨어있다. -> 단순히 한칸 뛰는게 아니라 두칸을 뛰어넘어가서 포도주를 마셔도 그것이 최대값일 수 있다.
   ex) 0 0 10 20 30
   나는 알고리즘을 한칸 또는 두칸만 점프하면 되게끔 짰는데 이렇게 될 경우 무조건 한칸 뒤, 또는 두칸 뒤에서 값을 선택해야한다. 따라서 이러한 문제를 해결해야한다.
2. 세칸을 점프하는 것을 추가하면, 원하지 않는 값을 또는 필요없는 값을 선택적으로 모두 점프할 수 있게 된다.

코드

#include<bits/stdc++.h>

using namespace std;

#define mine

int N;
int a[10001];
int dp[3][10001];

////// 틀린이유: 두칸 이상 건너뛰어도 더 클 수가 있다!
/// 나중에 다른사람 코드 보고 체크하기
int grape(int n, int c){
    if(n > N) return 0;
    if(c >= 3) return 0;
    int& ret = dp[c][n];
    if(ret != -1) return ret;

    ret = max(grape(n+1, c+1), grape(n+2, 1)) + a[n];
    ret = max(grape(n+3, 1)+a[n], ret);
    return ret;
}

int main(){
    ios_base::sync_with_stdio(false);
    // freopen("../input.txt","rt",stdin);

    memset(dp, -1, sizeof(dp));
    cin >> N;
    a[0] = 0;
    for(int i=1; i<=N; i++){
        cin >> a[i];
    }

    cout << grape(0,0) << endl;

    return 0;
}

소감

나는 두칸 이상을 넘어가도 된다는 것을 생각하지 못하였다. 비록 기본 뼈대는 맞았기 때문에 쉽게 문제를 해결했지만 나중에 다른 사람들이 푼 풀이도 보고 넣어놓도록 하겠다.