Leetcode 2938. 区分黑球和白球
桌子上有 n 个球,每个球的颜色不是黑色,就是白色。
给你一个长度为 n 、下标从 0 开始的二进制字符串 s,其中 1 和 0 分别代表黑色和白色的球。
在每一步中,你可以选择两个相邻的球并交换它们。
返回「将所有黑色球都移到右侧,所有白色球都移到左侧所需的 最小步数」。
不管输入字符串是何种状态,它的最终状态都是 0 在前面,1 在后面。
只需要考虑将白球移到前面即可。
class Solution {
public long minimumSteps(String s) {
int len = s.length();
long res = 0;
int whiteBallIndex = 0;
for (int i = 0; i < len; i++) {
if (s.charAt(i) == '0') {
res += i - whiteBallIndex;
whiteBallIndex++;
}
}
return res;
}
}