LeetCode 453——最小操作次数使数组元素相等（动态规划）

一、题目介绍

给定一个长度为 n 的 非空 整数数组，每次操作将会使 n - 1 个元素增加 1。找出让数组所有元素相等的最小操作次数。

示例：

输入：
[1,2,3]
输出：
3
解释：
只需要3次操作（注意每次操作会增加两个元素的值）：
[1,2,3]  =>  [2,3,3]  =>  [3,4,3]  =>  [4,4,4]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-moves-to-equal-array-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

本题采用动态规划的方法，首先将给定数组进行排序（从小到大排序），其次依次遍历排序后的数组，当访问第i个元素时计算diff = a[i]-a[i-1]，即a[i-1]和a[i]之间的差值。因为每次操作元素只能增加1，所以操作次数move += diff。按照题目要求每次将会操作n-1个元素，因此当遍历到第i个元素时，除了前i-1个元素要增加diff，后面的a[j] += diff（其中j > i && j < n）。实际上我们可以将a[j]的增幅，累积到move上面，当每次遍历到第i个元素时,先更新a[i] += move再计算diff，这样就不需要每次更新a[j]的值。遍历结束后move的值即为最小的操作次数。详见代码。

三、解题代码

class Solution {
public:
    int minMoves(vector<int>& nums) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        int diff = 0;
        int move = 0;
        for(int i = 1; i < n; ++i)
        {
            nums[i] += move;
            diff = nums[i] - nums[i-1];
            move += diff;
        }
        return move;
    }
};

四、解题结果