Top Secret Task(dp+滚动数组)

传送门： 题目

dp[i][j][k] 表示 考虑到第i个数 计算前j个数的和 进行了k次操作

则有

1. 若不把第i个数放入前j个数中

   dp[i][j][k] = dp[i-1][j][k]          
   

2. 若把第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;
}