|
| 1 | +--- |
| 2 | +title: 2938.区分黑球与白球 |
| 3 | +date: 2024-06-06 22:51:23 |
| 4 | +tags: [题解, LeetCode, 中等, 贪心, 双指针, 字符串] |
| 5 | +--- |
| 6 | + |
| 7 | +# 【LetMeFly】2938.区分黑球与白球:黑球要与后面每个白球交换一次 |
| 8 | + |
| 9 | +力扣题目链接:[https://leetcode.cn/problems/separate-black-and-white-balls/](https://leetcode.cn/problems/separate-black-and-white-balls/) |
| 10 | + |
| 11 | +<p>桌子上有 <code>n</code> 个球,每个球的颜色不是黑色,就是白色。</p> |
| 12 | + |
| 13 | +<p>给你一个长度为 <code>n</code> 、下标从 <strong>0</strong> 开始的二进制字符串 <code>s</code>,其中 <code>1</code> 和 <code>0</code> 分别代表黑色和白色的球。</p> |
| 14 | + |
| 15 | +<p>在每一步中,你可以选择两个相邻的球并交换它们。</p> |
| 16 | + |
| 17 | +<p>返回「将所有黑色球都移到右侧,所有白色球都移到左侧所需的 <strong>最小步数</strong>」。</p> |
| 18 | + |
| 19 | +<p> </p> |
| 20 | + |
| 21 | +<p><strong class="example">示例 1:</strong></p> |
| 22 | + |
| 23 | +<pre> |
| 24 | +<strong>输入:</strong>s = "101" |
| 25 | +<strong>输出:</strong>1 |
| 26 | +<strong>解释:</strong>我们可以按以下方式将所有黑色球移到右侧: |
| 27 | +- 交换 s[0] 和 s[1],s = "011"。 |
| 28 | +最开始,1 没有都在右侧,需要至少 1 步将其移到右侧。</pre> |
| 29 | + |
| 30 | +<p><strong class="example">示例 2:</strong></p> |
| 31 | + |
| 32 | +<pre> |
| 33 | +<strong>输入:</strong>s = "100" |
| 34 | +<strong>输出:</strong>2 |
| 35 | +<strong>解释:</strong>我们可以按以下方式将所有黑色球移到右侧: |
| 36 | +- 交换 s[0] 和 s[1],s = "010"。 |
| 37 | +- 交换 s[1] 和 s[2],s = "001"。 |
| 38 | +可以证明所需的最小步数为 2 。 |
| 39 | +</pre> |
| 40 | + |
| 41 | +<p><strong class="example">示例 3:</strong></p> |
| 42 | + |
| 43 | +<pre> |
| 44 | +<strong>输入:</strong>s = "0111" |
| 45 | +<strong>输出:</strong>0 |
| 46 | +<strong>解释:</strong>所有黑色球都已经在右侧。 |
| 47 | +</pre> |
| 48 | + |
| 49 | +<p> </p> |
| 50 | + |
| 51 | +<p><strong>提示:</strong></p> |
| 52 | + |
| 53 | +<ul> |
| 54 | + <li><code>1 <= n == s.length <= 10<sup>5</sup></code></li> |
| 55 | + <li><code>s[i]</code> 不是 <code>'0'</code>,就是 <code>'1'</code>。</li> |
| 56 | +</ul> |
| 57 | + |
| 58 | + |
| 59 | + |
| 60 | +## 解题方法:一次遍历 |
| 61 | + |
| 62 | +同色球交换无意义,因此所有的交换都是:前面的黑球与黑球后面的白球。 |
| 63 | + |
| 64 | +因此统计一下每个黑球后面有多少个白球即可。 |
| 65 | + |
| 66 | +怎么统计?使用一个变量记录当前遍历到的黑球数,遇到一个黑球就黑球数加一,遇到一个白球答案就加上黑球数。 |
| 67 | + |
| 68 | ++ 时间复杂度$O(len(s))$ |
| 69 | ++ 空间复杂度$O(1)$ |
| 70 | + |
| 71 | +### AC代码 |
| 72 | + |
| 73 | +#### C++ |
| 74 | + |
| 75 | +```cpp |
| 76 | +/* |
| 77 | +1010 |
| 78 | +1001 |
| 79 | +0101 |
| 80 | +0011 |
| 81 | +*/ |
| 82 | +typedef long long ll; |
| 83 | +class Solution { |
| 84 | +public: |
| 85 | + ll minimumSteps(string s) { |
| 86 | + ll ans = 0, times = 0; |
| 87 | + for (char c : s) { |
| 88 | + if (c == '1') { |
| 89 | + times++; |
| 90 | + } |
| 91 | + else { |
| 92 | + ans += times; |
| 93 | + } |
| 94 | + } |
| 95 | + return ans; |
| 96 | + } |
| 97 | +}; |
| 98 | +``` |
| 99 | +
|
| 100 | +#### Go |
| 101 | +
|
| 102 | +```go |
| 103 | +// package main |
| 104 | +
|
| 105 | +func minimumSteps(s string) int64 { |
| 106 | + ans, times := int64(0), 0 |
| 107 | + for _, c := range s { |
| 108 | + if c == '1' { |
| 109 | + times++ |
| 110 | + } else { // 必须和上一个大括号写在同一行 |
| 111 | + ans += int64(times) |
| 112 | + } |
| 113 | + } |
| 114 | + return ans |
| 115 | +} |
| 116 | +``` |
| 117 | + |
| 118 | +#### Java |
| 119 | + |
| 120 | +```java |
| 121 | +class Solution { |
| 122 | + public long minimumSteps(String s) { |
| 123 | + long ans = 0, times = 0; |
| 124 | + for (char c : s.toCharArray()) { |
| 125 | + if (c == '1') { |
| 126 | + times++; |
| 127 | + } |
| 128 | + else { |
| 129 | + ans += times; |
| 130 | + } |
| 131 | + } |
| 132 | + return ans; |
| 133 | + } |
| 134 | +} |
| 135 | +``` |
| 136 | + |
| 137 | +#### Python |
| 138 | + |
| 139 | +```python |
| 140 | +class Solution: |
| 141 | + def minimumSteps(self, s: str) -> int: |
| 142 | + ans, times = 0, 0 |
| 143 | + for c in s: |
| 144 | + if c == '1': |
| 145 | + times += 1 |
| 146 | + else: |
| 147 | + ans += times |
| 148 | + return ans |
| 149 | +``` |
| 150 | + |
| 151 | +> 同步发文于CSDN和我的[个人博客](https://blog.letmefly.xyz/),原创不易,转载经作者同意后请附上[原文链接](https://blog.letmefly.xyz/2024/06/06/LeetCode%202938.%E5%8C%BA%E5%88%86%E9%BB%91%E7%90%83%E4%B8%8E%E7%99%BD%E7%90%83/)哦~ |
| 152 | +> |
| 153 | +> Tisfy:[https://letmefly.blog.csdn.net/article/details/139511813](https://letmefly.blog.csdn.net/article/details/139511813) |
0 commit comments