https://codeforces.com/contest/1328/problem/F
首先把a数组处理成pair对（num,cnt），表示数字num有cnt个，然后按num升序排序离散化一下。
对于一个数x，若想使得小于x的数字都变成x，必须先把所有小于x的数变成x-1，然后再+1变成x。
同理，要使得大于x的数变成x，必须把所有大于x的数字变成x+1，然后再-1变成x。
以上是题意所要求的必须操作。
思路：
1. 用f[i]数组记录离散化后前i大的数字的总数，那么对于任意第i大数字，可以靠f[i]数组轻易求出大于num[i]的数字个数和小于num[i]的数字个数
2. 用一个树状数组维护
cur代表离散化后一共cur个不同的数字。
3.对于任意第i大数字，若k<=cnt[i]，则ans = 0，不用做任何操作，这是很显然的结论
4.对于任意第i大数字，若k>cnt[i]，设 d = k - cnt[i]，分为以下三种情况讨论：
(i) 比第i个数字大的数字有大于d个时，可以把大于num[i]的数字移动到num[i]+1的位置，这个过程的花费设为sum，再把d个数字移动到num[i]，花费sum+d，则即可满足条件
(ii)比第i个数字小的数字有大于d个时，可以把小于num[i]的数字移动到num[i]-1的位置， 这个过程的花费设为sum，再把d个数字移动到num[i]，花费sum+d，则即可满足条件
(iii)比第i个数字大的数字小于d个和比第i个数字小的数字小于d个同时发生，那么需要把大于num[i]和小于num[i]的数字全部分别移动到num[i]+1和num[i]-1，这整个过程花费记为sum3，再把d个数字移动到num[i]，总花费sum3+d，满足条件。
针对每个num[i]做以上三种操作，取min即可
1 #include<bits/stdc++.h>
2 using namespace std;
3 typedef long long ll;
4 const int maxn = 2e5+5;
5 ll c[maxn],a[maxn],f[maxn];
6 int n,k;
7 struct node{
8 ll cnt,num;
9 bool operator < (const node &b){
10 return num<b.num ;
11 }
12 }p[maxn],t[maxn];
13 ll lowbit(ll x){
14 return x& -x;
15 }
16 void add(ll x,ll k){
17 while(x<=n){
18 c[x] = c[x] + k;
19 x = x + lowbit(x);
20 }
21 }
22 ll getsum(ll x){
23 ll ans = 0;
24 while(x>=1){
25 ans +=c[x];
26 x = x-lowbit(x);
27 }
28 return ans;
29 }
30 int main(){
31 scanf("%d%d",&n,&k);
32 for(int i = 1;i<=n;i++) {
33 scanf("%lld",&p[i].num );
34 }
35 sort(p + 1,p + 1 +n);//离散化
36 t[1].num = p[1].num,t[1].cnt = 1,f[1] = 1;
37 int cur = 1;
38 for(int i = 2;i<=n;i++){
39 if(p[i].num !=t[cur].num ) t[cur+1].num = p[i].num ,t[cur+1].cnt = 1,cur++,f[cur]=f[cur-1]+1;
40 else t[cur].cnt ++,f[cur]++;//处理f函数和t数组
41 }
42 for(int i = 1;i<=cur;i++){
43 add(i,t[i].cnt *t[i].num );//维护前缀和
44 }
45 ll ans = 1e15+5;
46 for(int i = 1;i<=cur;i++){
47 if(k<=t[i].cnt ) {ans = 0;break;} //存在k>=cnt[i]直接break，ans = 0
48 else{
49 ll d = k - t[i].cnt ;//剩余需要d个移动到num[i]
50 if(f[cur]-f[i]>=d){
51 ll sum = getsum(cur) - getsum(i) - (f[cur]-f[i]) *(t[i].num+1);
52 ans = min(ans,sum+d);
53
54 }
55 if(f[i-1]>=d){
56 ll sum = f[i-1]*(t[i].num -1)- getsum(i-1);
57 ans = min(ans,sum+d);
58 }
59 if(f[cur]-f[i]<d && f[i-1]<d){
60 ll sum = getsum(cur) - getsum(i) - (f[cur]-f[i]) *(t[i].num+1);
61 sum = sum + f[i-1]*(t[i].num-1) - getsum(i-1);
62 ans = min(ans,sum+d);
63 }
64 }
65 }
66 cout<<ans;
67 return 0;
68 }
原文地址：https://www.cnblogs.com/AaronChang/p/12590292.html