美团笔试8.19 第五题 用例没错过,提交没对过!!!
小美拿到了一个数组,她每次可以进行如下操作:选择两个元素,一个加 1,另一个减 1。小美希望若干次操作后,众数的出现次数尽可能多。你能帮她求出最小的操作次数吗? 众数定义: 在一组数据中,出现次数最多的数据,是一组数据中的原数据,而不是相应的次数。一组数据中的众数不止一个,如数据2、3、-1、2、1、3中,2、3都出现了两次,它们都是这组数据中的众数
一顿操作猛如虎,一提交0%!!!!
#美团信息集散地#
class Main{
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
in.nextLine();
if(n == 1){
System.out.println(0);
return;
}
int[] nums = new int[n];
int sum = 0;
for(int i = 0; i < n; i++){
nums[i] = in.nextInt();
sum += nums[i];
}
Arrays.sort(nums);
int left = sum - nums[n - 1];
int right = sum - nums[0];
if(sum % n == 0){
System.out.println(adj2Target(nums,sum / n));
}else{
long res = adj2Target(Arrays.copyOfRange(nums, 0, nums.length - 1), left/(n - 1));
res = Math.min(res,adj2Target(Arrays.copyOfRange(nums, 0, nums.length - 1), left/(n - 1) + 1));
res = Math.min(res,adj2Target(Arrays.copyOfRange(nums, 1, nums.length), right/(n - 1)));
res = Math.min(res,adj2Target(Arrays.copyOfRange(nums, 1, nums.length), right/(n - 1) + 1));
System.out.println(res);
}
}
public static long adj2Target(int[] nums, int target){
long left = 0;
long right = 0;
for(int num : nums){
if(num > target)
right += (num - target);
else
left += (target - num);
}
return Math.max(left, right);
}
}
#美团信息集散地#