计数二进制子串
给定一个字符串 s,计算具有相同数量0和1的非空(连续)子字符串的数量,并且这些子字符串中的所有0和所有1都是组合在一起的。
重复出现的子串要计算它们出现的次数。
解题思路
我们可以将字符串 s 按照 0 和 1 的连续段分组,存在counts 数组中,例如 s=00111011,可以得到这样的 counts 数组:counts={2,3,1,2}。
这里counts 数组中两个相邻的数一定代表的是两种不同的字符。假设counts 数组中两个相邻的数字为 u 或者 v,它们对应着 u 个 0 和 v 个 1,或者 u个 1 和 v 个 0。它们能组成的满足条件的子串数目为 min{u,v},即一对相邻的数字对答案的贡献。
代码
class Solution:
def countBinarySubstrings(self, s: str) -> int:
count = [1]
j = 0
for i in range(1, len(s)): # 统计连续的频数
if s[i] == s[i-1]:
count[j] += 1
else:
count.append(1)
j += 1
res = 0
for k in range(1, len(count)):
res += min(count[k], count[k-1]) # 取相邻频数的最小值
return res
时间复杂度:O(N)
空间复杂度:O(N)