LeetCode 1769 移动所有球到每个盒子所需的最小操作数[数组] HERODING的LeetCode之路

Posted 2022-12-15 HERODING23

解题思路：
一眼看这道题就能联想到前缀和，但是使用前缀和的话得到的只是前$i$个盒子中小球的总和，没有步数信息，那么就魔改一下，遍历的时候记录前$i$个盒子中小球的步数之和pre，以及小球的个数之和count，那么对于当前位置，前$i$个小球到达所需要的步数为pre+count，然后将pre更新为pre+count，这代表着再往下走，前i个小球又要多走count步，正向遍历结束再反向遍历执行同样的操作，这样就得到每个位置前后小球达到所需要的步数总和，代码如下：

class Solution 
public:
    vector<int> minOperations(string boxes) 
        int n = boxes.size();
        vector<int> res(n, 0);
        // 前面球到后面位置所需要的步数
        int pre = 0, count = 0;
        for(int i = 1; i < n; i ++) 
            if(boxes[i - 1] == '1') 
                count ++;
            
            res[i] += pre + count;
            pre += count;
        
        // 后面球到前面位置所需要的步数
        pre = 0, count = 0;
        for(int i = n - 2; i >= 0; i --) 
            if(boxes[i + 1] == '1') 
                count ++;
            
            res[i] += pre + count;
            pre += count;
        
        return res;
    
;