（dp）2020牛客寒假算法基礎集訓營2H.施魔法

2020牛客寒假算法基礎集訓營2H.施魔法

思路：

比賽時推了一個n²的dp，一看肯定TLE/MLE。
先排序。
dp[i]表示選前i個花費魔法最小值。
dp[i]=min(dp[j-1]+a[i]-a[j])=min(dp[j-1]-a[j])+a[i] (1≤j≤i-k+1)。維護dp[j-1]-a[j]的最小值。

代碼：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
const int N=3e5+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
int a[N],dp[N];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,k,i;
	cin>>n>>k;
	for(i=1;i<=n;i++)
		cin>>a[i];
	sort(a+1,a+n+1);
	int t=-a[1];
	for(i=1;i<k;i++)
		dp[i]=inf;
	for(i=k;i<=n;i++)
	{
		dp[i]=t+a[i];
		t=min(t,dp[i-k+1]-a[i-k+2]);
	}
	cout<<dp[n]<<endl;
	return 0;
}