LeetCode 517 超级洗衣机[数学 逻辑] HERODING的LeetCode之路

Posted 2021-10-19 HERODING23

解题思路：
虽然是一道困难题，但是代码实现起来并不复杂，关键还是在于思路。首先，第一步是统计所有的衣服件数，如果不满足均分条件直接返回-1，接着计算均值avg，并且统计洗衣机内衣服数量的前缀和，每遍历到一台洗衣机，它需要转移衣服的个数最小值是machine[i] - avg，但是这是在前i个洗衣机内的衣服自给自足的情况（即前i个洗衣机的衣服的均值正好为avg），否则需要往后移或者往前移（多了的情况和少了的情况），这个时候移动次数考虑最大值，代码如下：

class Solution {
public:
    int findMinMoves(vector<int>& machines) {
        int ans = 0, sum = 0, presum = 0;
        for (int i = 0; i < machines.size(); i++)
            sum += machines[i];

        if (sum % machines.size() != 0)
            return -1;

        int avg = sum / machines.size();

        for (int i = 0; i < machines.size(); i++) {
            presum += machines[i];
            ans = max(ans, max(machines[i] - avg, abs(presum - avg * (i + 1))));
        }
        return ans;
    }
};