2020牛客寒假算法基础集训营2 H.施魔法

H.施魔法

题目链接-施魔法


解题思路
DP，dp[i]表示用掉前 i 个元素的最小代价；

• 先将元素按照能量值从小到大排序
• 因为至少要取k 个，所以前面k−1 个是不能直接取到的，所以 前i−1个的dp值为INF
• 当我们计算第i个元素的时候，它肯定是最大元素，第i-1个元素就不是最大的了，所以维护min{dp[j]−a[j+1]}
  (j∈[0,i−k])即可,可得到状态转移方程dp[i]=min(dp[i-1]-a[i-1]+a[i],dp[i-k]-a[i-k+1]+a[i]);

附上代码

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f;
const int N=3e5+5;
typedef long long ll;
typedef pair<int,int> PII;
int a[N],dp[N];
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	
	int n,k;
	cin>>n>>k;
	memset(dp,INF,sizeof dp);
	for(int i=1;i<=n;i++)
		cin>>a[i];
	sort(a+1,a+n+1);
	dp[k]=a[k]-a[1];
	for(int i=k+1;i<=n;i++){
		dp[i]=min(dp[i-1]-a[i-1]+a[i],dp[i-k]-a[i-k+1]+a[i]);
	}
	cout<<dp[n]<<endl;
	return 0;
}