【教3妹学编程-算法题】区分黑球与白球

2023-11-20  本文已影响0人  程序员小2
图片来源网络

2哥 : 3妹,有没有看到AI圈爆炸新闻:突发,OpenAI政变再反转!阿尔特曼逼宫失败,董事会又换了新CEO
3妹:啊?他不是前两天刚被裁员嘛?
2哥 : 是啊, 简直堪比宫斗剧了,哈哈~吃瓜吃瓜
3妹:宫斗剧还有黑白角色,你说他们这些股东谁是黑谁是白?
2哥 : 这个不好说,普通程序员只要吃瓜就行了,哪管那么多。
3妹:说到黑白,我看到一个相关的题目,让我来考考你吧~

吃瓜

题目:

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

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

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

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

示例 1:

输入:s = "101"
输出:1
解释:我们可以按以下方式将所有黑色球移到右侧:

输入:s = "100"
输出:2
解释:我们可以按以下方式将所有黑色球移到右侧:

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

提示:

1 <= n == s.length <= 10^5
s[i] 不是 '0',就是 '1'。

思路:

思考

类似冒泡排序,

对于每个 0,它左边有多少个 1,就移动多少次。

所以一边遍历 s,一边统计 1 的个数 cnt1 ,遇到 0 就把 cnt1加入答案。

java代码:

class Solution {
    public long minimumSteps(String s) {
        long ans = 0;
        int cnt1 = 0;
        for (char c : s.toCharArray()) {
            if (c == '1') {
                cnt1++;
            } else {
                ans += cnt1;
            }
        }
        return ans;
    }
}

上一篇 下一篇

猜你喜欢

热点阅读