LeetCode-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 <= 10^5
s[i] 不是 ‘0’,就是 ‘1’。
解题思路一:贪心
class Solution:
def minimumSteps(self, s: str) -> int:
ans = cnt1 = 0
for c in s:
if c == '1':
cnt1 += 1
else:
ans += cnt1
return ans
时间复杂度:O(n)
空间复杂度:O(1)
解题思路二:一次遍历统计’1’的个数,找’0’后累加左边的’1’的个数
想不到贪心怎么办?背吧。
class Solution:
def minimumSteps(self, s):
ans, sum = 0, 0
for i in range(len(s)):
if s[i] == '1':
sum += 1
else:
ans += sum
return ans
时间复杂度:O(n)
空间复杂度:O(1)
解题思路三:
时间复杂度:O(n)
空间复杂度:O(1)
♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠