leetcoed 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
解释：所有黑色球都已经在右侧。

C代码：

long long minimumSteps(char *s) {

    long long ans = 0;

    int sum = 0;

    for (int i = 0; i < strlen(s); i++) {

        if (s[i] == '1') {

            sum++;

        } else {

            ans += sum;

        }

    }

    return ans;

}

解析：

for (int i = 0; i < strlen(s); i++)：开始一个 for 循环，循环变量 i 从 0 开始，直到字符串 s 的长度减 1。strlen(s) 函数计算字符串 s 的长度。

if (s[i] == '1')：检查当前索引 i 处的字符是否是 '1'。

sum++;：如果当前字符是 '1'，则将 sum 增加 1。这表示我们遇到了一个新的连续的 '1'。

else：如果当前字符不是 '1'（即它是 '0'），则执行ans += sum;。

ans += sum;：将当前的 sum 值加到 ans 上。因为每次遇到 '0' 时，之前的所有连续 '1' 都需要被交换到字符串的末尾，所以交换次数等于连续 '1' 的数量。

笔记：

该算法的巧妙之处在于，只需要遍历数组一次，每一个元素都会进行判断该元素是否为 '1'，若是，则 sum 便自增一次。当遍历到第 n 个元素是 '0' 时，则将 '0' 移动到 '1' 前面需要和前面的每个 '1' 都进行交换。

————————————————

                            版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
                        
原文链接：https://blog.csdn.net/weixin_45048860/article/details/139481182