DP-牛客寒假集训营2- 施魔法

DP-牛客寒假集训营2- 施魔法

题目：

题意：
$输入n，k。$
$输入n个数a_1,a_2,...,a_n。$
$将这n个数分成几个集合，每个集合中至少包含k个元素。$
每 个 集 合 S 的 代 价 为 m a x { a i − a j } , ( a i , a j ∈ S ) 。 每个集合S的代价为max \{a_i-a_j\},(a_i,a_j∈S)。 每个集合S的代价为max{ai​−aj​},(ai​,aj​∈S)。
$求所有集合代价总和的最小值。$

题解：
$要使得总代价最小，那么每个集合中最大值与最小值的差就要尽量小。$

$容易证明最优解必然是在连续有序的区间中取得的，因此首先将n个数排序。$

$在前i个元素中得到的最小代价与在前i-1个元素中得到的最小代价是相关的。$

$我们设dp[i]代表考虑前i个元素的最小代价,显然当i<k时,dp[i]是没有意义的。$

$则dp[i]=min(dp[j-1]+a[i]-a[j])=min(dp[j-1]-a[j])+a[i],(j\in [1,i-k+1])。$

$即考虑区间[1,i]的元素的最小代价=min(区间[0,j-1]的元素的最小代价+区间[j,i]的代价，即最大值a[i]-最小值a[j]),(j\in [1,i-k+1])。$

$最后等价于求min(dp[j-1]-a[j]),(j\in [1,i-k+1])，即dp[j-1]-a[j]前缀和的最小值$

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const ll N=3e5+5;
int n,k;
ll a[N],dp[N];//  dp[i]:使用前i个元素的最小代价
ll pre;
//dp[i]=min(dp[j-1]+a[i]-a[j])=min(dp[j-1]-a[j])+a[i]
//     j∈[1,i-k+1]
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);

    sort(a+1,a+n+1);

    for(int i=1;i<k;i++) dp[i]=inf;//前k-1项无法满足题意，初始化为inf
    pre=dp[0]-a[1];  //pre即dp[i-1]-a[i]前缀最小值
    
    for(int i=k;i<=n;i++)
    {
        dp[i]=a[i]+pre; //
        pre=min(pre,dp[i-k+1]-a[i-k+2]);
    }
    cout<<dp[n]<<endl;

    return 0;
}