Manthan, Codefest 18 (rated, Div. 1 + Div. 2) B. Reach Median (cf 1037 B)

题目：Reach Median

题意：给出一个长度为n的序列和一个整数k，求最少需要多少次操作可以把序列的中位数变成k。一次操作定义为将序列中的任意一个数+1或-1。

思路：
经过观察可以知道，不改变序列元素之间的相对大小是更优的选择。
先对序列进行排序。
如果当前中位数小于k，那么就把它加到k，为了保证序列的相对大小不变，需要把它后面小于k的数都加到k。大于k时同理。

代码：

#include<bits/stdc++.h>
using namespace std;

#define ll long long
#define maxn 200000

int n,k;
int a[maxn+5];

int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
    }
    sort(a+1,a+n+1);
    int mid=(n+1)/2;
    if(a[mid]==k) {
        printf("0");
        return 0;
    } else if(a[mid]<k) {
        ll ans=0;
        for(int i=mid;i<=n;i++) {
            if(a[i]<k) ans+=k-a[i]; 
        }
        printf("%I64d",ans);
    } else if(a[mid]>k) {
        ll ans=0;
        for(int i=1;i<=mid;i++) {
            if(a[i]>k) ans+=a[i]-k; 
        }
        printf("%I64d",ans);
    }

    return 0;
}