leetcode_696 计数二进制子串

1,125 阅读1分钟

要求

给定一个字符串 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”字符。

核心代码

class Solution:
    def countBinarySubstrings(self, s: str) -> int:
        count = 0
        for i in range(len(s) - 1):
            left,right = s[i],s[i + 1]
            left_idx,right_idx = i,i + 1
            if left != right:
                while 0 <= left_idx <= right_idx < len(s) and s[left_idx] == left and s[right_idx] == right:
                    count += 1
                    left_idx,right_idx = left_idx - 1,right_idx + 1
        return count 

image.png

解题思路:我们使用一个长度为2的滑窗遍历整个字符串,当我们遇到了left != rihgt的时候我们在当前窗口向左和向右扩展,如果还是满足左侧和左侧的相同,右侧和右侧的相同,这样的话就能不断的得到子串,思路很好,值得借鉴。