[BOJ 2156 ] - 포도주 시식

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

일렬로 있는 포도주를 일정 규칙에 따라서 마실 때 최대로 마실 수 있는 양을 구하는 DP문제이다. 

#include <iostream>
#include <cstdio>
#include <cstring>
 
 
using namespace std;
int n;
int dp[3][22222];
int arr[22222];
int main(){
    memset(dp,0,sizeof(dp));
    memset(arr,0,sizeof(arr));
    scanf("%d" , &n);
    
    for(int i=1;i<=n;i++){
        scanf("%d" , &arr[i]);
    }
    
    dp[1][1] = arr[1];
    dp[2][1] = arr[1];
    dp[1][2] = arr[2];
    dp[2][2] = arr[1]+arr[2];
    
    for(int i=3;i<=n;i++){
        dp[1][i] = max(dp[2][i-1] ,max(dp[2][i-2]+arr[i] ,dp[1][i-2]+ arr[i]));
        dp[2][i] = dp[1][i-1] + arr[i];
    }
    
    int ans = 0;
    
    for(int i=1;i<=n;i++){
        if(ans< dp[1][i])
            ans = dp[1][i];
        if(ans< dp[2][i])
            ans = dp[2][i];
    }
    
    printf("%d\n" , ans);
    
    
    return 0;
}

처음에는 2차원 DP로 구하면 되겠네 하고 아무생각없이 구현하였고 , 되긴 하였으나 뭔가 깔끔하지 않았고 ... 

다른사람들의 코드를 보니 1차원 DP로도 충분히 구현이 가능하였다. 점화식을 새울때 3가지로 나누어서 구할 수 있는데 

1) 0번 연속하여 음료를 마시는 경우 --> arr[i] 를 마시지 않음 [dp[i] = dp[i-1]]

2) 1번 연속하여 음료를 마시는 경우 --> arr[i-1]을 마시지 않은경우 [dp[i] = dp[i-2]+arr[i]]

3) 2번 연속하여 음료를 마시는 경우 --> arr[i]와 arr[i-1]을 마시고 그 전껀 먹지 않은 경우 [ dp[i] = dp[i-3] + arr[i-1] + arr[i]]

로 점화식을 세워서 구현하면 더 깔끔하게 구현할 수 있다. 

#include <iostream>
#include <cstdio>
#include <cstring>
 
using namespace std;
 
int dp[22222];
int arr[22222];
int n,ans;
 
int main(){
    scanf("%d" , &n);
    ans = 0;
    
    for(int i=1;i<=n;i++)
        scanf("%d" , &arr[i]);
    
    dp[0] = 0;
    dp[1] = arr[1];
    dp[2] = arr[1]+arr[2];
    
    for(int i=3;i<=n;i++)
        dp[i] = max(dp[i-1] , max(dp[i-3]+arr[i-1]+arr[i] , dp[i-2]+arr[i]));
    
    for(int i=1;i<=n;i++)
        if(dp[i] > ans)
            ans = dp[i];
    
    printf("%d\n" , ans);
    
    return 0;
}