LeetCode：2938. 区分黑球与白球（贪心 Java）

目录

2938. 区分黑球与白球

题目描述：

实现代码与解析：

贪心

原理思路：

---

2938. 区分黑球与白球

题目描述：

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

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

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

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

示例 1：

输入：s = "101"
输出：1
解释：我们可以按以下方式将所有黑色球移到右侧：
- 交换 s[0] 和 s[1]，s = "011"。
最开始，1 没有都在右侧，需要至少 1 步将其移到右侧。

示例 2：

输入：s = "100"
输出：2
解释：我们可以按以下方式将所有黑色球移到右侧：
- 交换 s[0] 和 s[1]，s = "010"。
- 交换 s[1] 和 s[2]，s = "001"。
可以证明所需的最小步数为 2 。

示例 3：

输入：s = "0111"
输出：0
解释：所有黑色球都已经在右侧。

提示：

• 1 <= n == s.length <= 105
• s[i] 不是 '0'，就是 '1'。

实现代码与解析：

贪心

class Solution {
    public long minimumSteps(String s) {

        int n = s.length();

        char[] cs = s.toCharArray();
        long res = 0;
        int cur = 0;
        for (int i = 0; i < n; i++) {
            if (cs[i] == '1') cur++;
            else {
                res += cur;
            }
        }
        return  res;
    }
}

原理思路：

        因为1都要放到0的右边，所以遍历的时候记录1的个数，直到遇到0就进行交换，交换次数就是之前出现的1的个数。

我一开始写的丑陋代码：

        我一开始想的是最早遇到的0和最早遇到的1交换。太小丑了我，不过倒是能过肯定。

    public long minimumSteps(String s) {

        int n = s.length();

        Queue<Integer> q = new ArrayDeque<>();
        char[] cs = s.toCharArray();
        long res = 0;
        for (int i = n - 1; i >= 0; i--) {
            char t = cs[i];
            if (t == '0') {
                q.offer(i);
            } else {
                if (!q.isEmpty()) {
                    int idx = q.peek();
                    q.poll();
                    res += idx - i;
                    cs[i] = cs[i + 1];
                    i++;

                }
            }
        }
        return  res;
    }