LeetCode—最小移动次数使数组元素相等(数学)

最小移动次数使数组元素相等(简单)

2020年7月20日

题目来源:力扣

解题
用数学方法可解,需要推导公式
sum为数组总和
min_val为数组最小的数
moves为移动次数
n为数组元素个数

假设目前数组总和为sum,我们需要移动次数为moves,那么整体数组总和将会增加moves* (n - 1),这里的n为数组长度,最后数组所有元素都相等为x,于是有:

sum + moves* (n - 1) = x * n (1)

我们再设数组最小的元素为min_val,moves= x - min_val​,即 ​x = moves+ min_val​带入(1)得:

moves= sum - min_val * n​

class Solution {
    public int minMoves(int[] nums) {
        int sum = 0;
        int min_val = Integer.MAX_VALUE;
        for(int i:nums){
            sum += i;
            min_val = Math.min(min_val,i);
        }
        return sum - min_val * nums.length;
    }
}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_41541562/article/details/107468310