选择数字[单调队列+DP]

传送门

我们发现直接dp不好做 

其实选的最大值就是总的值减去选的最小值

我们选最小值要保证任意k个选一个,这就是比较显然的dp了

f[i]=min(f[j])+a[i](i-j-1<=k)

单调队列维护一下就好


#include<bits/stdc++.h>
#define N 100050
#define LL long long
#define inf 1000000000000000
using namespace std;
int n,k,a[N],q[N]; LL f[N],tot,ans=inf;
int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) 
		scanf("%d",&a[i]),tot+=a[i];
	int l=1,r=0; q[++r]=0;
	for(int i=1;i<=n;i++){
		while(l<=r && i-q[l]>k+1) l++;
		f[i] = (LL)f[q[l]] + a[i];
		while(l<=r && f[i]<f[q[r]]) r--;
		q[++r] = i;
	}
	for(int i=n-k;i<=n;i++) ans = min(ans,f[i]);
	printf("%lld",tot-ans); return 0;
}

猜你喜欢

转载自blog.csdn.net/sslz_fsy/article/details/84678895