第十六届浙江大学宁波理工学院程序设计大赛 D 雷顿女士与分队hard version(dp)
作者:互联网
题意
链接:https://ac.nowcoder.com/acm/contest/2995/D
来源:牛客网
卡特莉接到来自某程序设计竞赛集训队的邀请,来为他们进行分队规划。
现在集训队共有n名选手,选手们的实力可以用一个整数来表示。
当若干个选手被分到了一队,队内会因为实力的不平衡而产生矛盾。
具体的来说,我们用矛盾因数来量化矛盾的大小,一个队的矛盾因数为队内成员实力的最大值减去最小值。
现在我们需要将n名选手恰好分为n/k向下取整队,并且每个队最少有k个人。
请你帮卡特莉回答一下他们分出的队的矛盾因数总和最小是多少。
思路
设dp[i]为从1分配到i的矛盾因数总和最小值。先对数组从小到大排序。当前的答案可以由dp[i-k]转移过来,即i-k+1~i(正好k人)分为一队;也可以由dp[i-1]转移过来,意味着i要加入到i-1这一队。
两种情况写成方程为:dp[i]=min(dp[i-k]+a[i]-a[i-k+1],dp[i-1]+a[i]-a[i-1])。
但是要注意,在i%k==0的时候,dp[i]只能由前者转移过来 ,因为要保证每个队都至少有k个人,那么每次找到k个人就可以算一下每个队的矛盾和。
代码
#include<bits/stdc++.h> using namespace std; #define inf 0x3f3f3f3f #define ll long long const int N=200005; const int mod=1e9+7; const double eps=1e-8; const double PI = acos(-1.0); #define lowbit(x) (x&(-x)) int a[N],dp[N]; int main() { std::ios::sync_with_stdio(false); int t; cin>>t; while(t--) { int n,k; cin>>n>>k; for(int i=1; i<=n; i++) { cin>>a[i]; } sort(a+1,a+1+n); for(int i=1; i<=n; i++) { if(i<=k+n%k) { dp[i]=a[i]-a[1]; } else dp[i]=inf; } for(int i=k+n%k+1; i<=n; i++) { if(i-k>=0) dp[i]=dp[i-k]+a[i]-a[i-k+1]; if(i%k) dp[i]=min(dp[i],dp[i-1]+a[i]-a[i-1]); // cout<<i<<" "<<dp[i]<<endl; } cout<<dp[n]<<endl; } return 0; }
标签:const,int,hard,矛盾,因数,version,雷顿,dp,define 来源: https://www.cnblogs.com/mcq1999/p/12006696.html