P5424 [USACO19OPEN] Snakes G

典型的资源分配类 DP。

那么这类 DP 怎么做呢？一般分三重循环。

第一重： 循环 i 表示前 i 个物品

第二重： 循环 j 表示现在要分配第 j组物品

第三重： 循环 k 表示将编号 k+1 到 i 的物品分配给第 j组

将sum进行前缀和处理

#include<bits/stdc++.h>
using namespace std;

const int N=410,INF=0x3f3f3f3f;
int n,k,a[N],s[N],f[N][N],mx[N][N];

int main(){
	scanf("%d %d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
	for(int i=1;i<=n;i++)
		for(int j=i;j<=n;j++)
			mx[i][j]=max(mx[i][j-1],a[j]);
	memset(f,0x3f,sizeof(f));
	for(int i=0;i<=n;i++) f[i][0]=mx[1][i]*i-s[i];
	for(int i=1;i<=n;i++){
		for(int j=1;j<=k;j++){
			for(int t=1;t<=i;t++){
				f[i][j]=min(f[i][j],f[t-1][j-1]+mx[t][i]*(i-t+1)-(s[i]-s[t-1]));
			}
		}
	}
	int ans=INF;
	for(int i=0;i<=k;i++) ans=min(ans,f[n][i]);
	printf("%d\n",ans);
	return 0;
}