给定一个字符串 s,计算具有相同数量0和1的非空(连续)子字符串的数量,并且这些子字符串中的所有0和所有1都是组合在一起的。
重复出现的子串要计算它们出现的次数。
示例 1 :
输入: “00110011”
输出: 6
解释: 有6个子串具有相同数量的连续1和0:“0011”,“01”,“1100”,“10”,“0011” 和 “01”。
请注意,一些重复出现的子串要计算它们出现的次数。
另外,“00110011”不是有效的子串,因为所有的0(和1)没有组合在一起。
示例 2 :
输入: “10101”
输出: 4
解释: 有4个子串:“10”,“01”,“10”,“01”,它们具有相同数量的连续1和0。
链接:https://leetcode-cn.com/problems/count-binary-substrings
方法:我们可以将字符串 s 按照 0 和 1 的连续段分组,存在counts 数组中,例如 s = 00111011,可以得到这样的 \rm counts 数组: counts = {2, 3, 1, 2}。
这个实现的时间复杂度和空间复杂度都是 O(n)O(n)。
改进:对于某一个位置 i,其实我们只关心 i - 1 位置的counts 值是多少,所以可以用一个 last 变量来维护当前位置的前一个位置,这样可以省去一个 counts 数组的空间。
这个实现的时间复杂度和空间复杂度分别为 O(n)O(1)。
链接:https://leetcode-cn.com/problems/count-binary-substrings/solution/ji-shu-er-jin-zhi-zi-chuan-by-leetcode-solution/
class Solution:
def countBinarySubstrings(self, s: str) -> int:
ans = 0
last = 0
len_s = len(s)
i = 0
while i <= len_s - 1:
now_num = s[i]
count = 0
while (i <= len_s - 1) and now_num == s[i]:
i = i + 1
count = count + 1
ans += min(count, last)
last = count
return ans