LeetCode 1769. 移动所有球到每个盒子所需的最小操作数（前缀和）

示例 1：
输入：boxes = "110"
输出：[1,1,3]
解释：每个盒子对应的最小操作数如下：
1) 第 1 个盒子：将一个小球从第 2 个盒子移动到第 1 个盒子，需要 1 步操作。
2) 第 2 个盒子：将一个小球从第 1 个盒子移动到第 2 个盒子，需要 1 步操作。
3) 第 3 个盒子：将一个小球从第 1 个盒子移动到第 3 个盒子，需要 2 步操作。
将一个小球从第 2 个盒子移动到第 3 个盒子，需要 1 步操作。共计 3 步操作。

示例 2：
输入：boxes = "001011"
输出：[11,8,5,4,3,4]
 
提示：
n == boxes.length
1 <= n <= 2000
boxes[i] 为 '0' 或 '1'

class Solution {
public:
    vector<int> minOperations(string boxes) {
        int n = boxes.size(), b = 0;
        vector<int> ans(n), temp(n);
        for(int i = 0; i < n; i++)
        {
            ans[i] += b + (i > 0 ? ans[i-1] : 0);
            // 移动到前一个位置的步数 + 前面盒子的个数
            if(boxes[i]=='1')
                b++;//前面的盒子个数
        }
        b = 0;
        for(int i = n-1; i >= 0; i--)
        {
            temp[i] += b + (i < n-1 ? temp[i+1] : 0);
            if(boxes[i]=='1')
                b++;
            ans[i] += temp[i];//两侧的步数合并
        }
        return ans;
    }
};