Day 14:2938. 区分黑球和白球

136 阅读1分钟

Leetcode 2938. 区分黑球和白球

桌子上有 n 个球,每个球的颜色不是黑色,就是白色。

给你一个长度为 n 、下标从 0 开始的二进制字符串 s,其中 1 和 0 分别代表黑色和白色的球。

在每一步中,你可以选择两个相邻的球并交换它们。

返回「将所有黑色球都移到右侧,所有白色球都移到左侧所需的 最小步数」。

image.png

不管输入字符串是何种状态,它的最终状态都是 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;
    }
}