$O(nm)$
간단한 동적계획법 문제이다.
dp[i]: 1~i 번 오렌지를 포장하는 최소 비용
dp[0]=0
dp[i]=min(j=i-1 ~ j>=0 & j>=i-m )(dp[j] + k +(max(a[j+1 ~ i])-min(a[j+1 ~ i]))*(i-j))
#include<cstdio> #include<algorithm> using namespace std; long long dp[20001]; int n, m, k, a[20001]; int main() { scanf("%d %d %d", &n, &m, &k); for (int i = 1; i <= n; i++) { scanf("%d", a + i); int mini = a[i], maxi = a[i]; dp[i] = 1e15; for (int j = i - 1; j >= 0 && j >= i - m; j--) { dp[i] = min(dp[i], dp[j] + k + (long long)(maxi - mini)*(i - j)); mini = min(mini, a[j]); maxi = max(maxi, a[j]); } } printf("%lld", dp[n]); return 0; }
댓글 없음 :
댓글 쓰기