给定一个字符串 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。
注意:
- s.length 在1到50,000之间。
- s 只包含“0”或“1”字符。
字符分组
我们可以将字符串 sss 按照 000 和 111 的连续段分组,存在 counts\rm countscounts 数组中,例如 s=00111011s = 00111011s=00111011,可以得到这样的 counts\rm countscounts 数组:counts={2,3,1,2}{\rm counts} = \{2, 3, 1, 2\}counts={2,3,1,2}。
这里 counts\rm countscounts 数组中两个相邻的数一定代表的是两种不同的字符。假设 counts\rm countscounts 数组中两个相邻的数字为 uuu 或者 vvv,它们对应着 uuu 个 000 和 vvv 个 111,或者 uuu 个 111 和 vvv 个 000。它们能组成的满足条件的子串数目为 min{u,v}\min \{ u, v \}min{u,v},即一对相邻的数字对答案的贡献。
对于某一个位置 iii,其实我们只关心 i−1i - 1i−1 位置的 counts\rm countscounts 值是多少,所以可以用一个 last\rm lastlast 变量来维护当前位置的前一个位置,这样可以省去一个 counts\rm countscounts 数组的空间。
Code
def countBinarySubstrings(self, s: str) -> int:
ans, cur, last, length = 0, 0, 0, len(s)
while cur < length:
c, count = s[cur], 0
while cur < length and s[cur] == c:
cur += 1
count += 1
ans += min(count, last)
last = count
return ans
复杂度分析
- 时间复杂度:O(n)O(n)O(n)。
- 空间复杂度:O(1)O(1)O(1)。