当前位置: 代码迷 >> 综合 >> Manthan, Codefest 18 (rated, Div. 1 + Div. 2) B. Reach Median (cf 1037 B)
  详细解决方案

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

热度:30   发布时间:2023-12-06 08:05:26.0

题目: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;
}
  相关解决方案