传送门: 题目
dp[i][j][k] 表示 考虑到第i个数 计算前j个数的和 进行了k次操作
则有
-
若不把第i个数放入前j个数中
dp[i][j][k] = dp[i-1][j][k]
-
若把第i个数放入前j个数中
至少需要把第i个移到第j个,即进行i-j次操作 if(k >= i-j) dp[i][j][k] = dp[i-1][j-1][k-(i-j)]
然而如果直接开数组,则要开dp[151][151][151*151]这么大 这显然是不可取的
又因为 dp[i][ ][ ] 始终只和 dp[i-1][ ][ ] 有关,所以考虑滚动数组
即用 dp[i%2][ ][ ] 表示 dp[i][ ][ ]
话不多说,上代码:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int N,K,S;
ll dp[2][155][155*155];
ll a[155];
int main(){
scanf("%d%d%d",&N,&K,&S);
for(int i=1;i<=N;i++)
scanf("%d",&a[i]);
memset(dp,inf,sizeof(dp));
dp[0][0][0]=0;
for(int i=1;i<=N;i++){
memset(dp[i%2][0],0,sizeof(dp[i%2][0]));
for(int j=1;j<=i;j++){
for(int k=0;k<=i*j;k++){
dp[i%2][j][k]=dp[(i-1)%2][j][k];
if(k>=i-j)
dp[i%2][j][k]=min(dp[i%2][j][k],dp[(i-1)%2][j-1][k-(i-j)]+a[i]);
}
}
}
ll ans=inf;
for(int k=0;k<=min(S,N*(N-1)/2);k++)
ans=min(dp[N%2][K][k],ans);
printf("%lld\n",ans);
return 0;
}