给定一个大小为n的非空整数数组,找出使所有数组元素相等所需的最小移动数,其中移动将n-1元素增加1

题干

leetcode 453. Minimum Moves to Equal Array Elements 三种语言解

Example:

Input:
[1,2,3]

Output:
3

Explanation:
Only three moves are needed (remember each move increments two elements):

[1,2,3] => [2,3,3] => [3,4,3] => [4,4,4]

简单讲就是把数组中的数全部弄成一样,方法是每次给任意n-1个数自增1

思路

这个其实是数学题,想通了就简单了
假设原来的数组元素和为sum,有n个元素,需要自增m次,原来的最小值为min,数字一样后数组的值均为x

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

另外,发现最小的数min一定自增了m次,因为每次自增时只自增n-1个数,剩下不自增的数一定是当前数组中最大的,而原数组中最小的数在过程结束前都一定不是最大的
所以
min+m=x
将x带入第一个式子得
m=sum-min * n

(盗一个图)

代码

java

class Solution {
    public int minMoves(int[] nums) {
        int n=nums.length;
        int min=nums[0];
        int sum=0;
        for(int x:nums){
            if(x<min)min=x;
            sum+=x;
        }
        return sum-min*n;
    }
}

python(是不是很简洁!也很美)

class Solution:
    def minMoves(self, nums):
        return sum(nums)-min(nums)*len(nums)          

C++

class Solution {
public:
    int minMoves(vector<int>& nums) {
        int sum=0;
        int min=nums[0];
        int n=nums.size();
        for(int x:nums){
            if(min>x)min=x;
            sum+=x;
        }
        return sum-n*min;
    }
};
全部评论

相关推荐

Pandaileee:校友加油我现在也只有一个保底太难了
点赞 评论 收藏
分享
Hello_WordN:咱就是说,除了生命其他都是小事,希望面试官平安,希望各位平时也多注意安全
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务