解题思路:
一眼看这道题就能联想到前缀和,但是使用前缀和的话得到的只是前
i
i
i个盒子中小球的总和,没有步数信息,那么就魔改一下,遍历的时候记录前
i
i
i个盒子中小球的步数之和pre,以及小球的个数之和count,那么对于当前位置,前
i
i
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;
}
};