题目: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 200000int 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;
}