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

Python 解答:
1.暴力查找

class Solution:
    def countBinarySubstrings(self, s: str) -> int:
        total = 0
        for i in range(1, len(s)):
            if s[i-1] != s[i]:
                j = i-1
                k = i
                while j >= 0 and k < len(s):
                    if s[j] == s[i-1] and s[k] == s[i]:
                        total += 1
                    else:
                        break
                    j -= 1
                    k += 1
        return total

2.线性扫描

class Solution:
    def countBinarySubstrings(self, s: str) -> int:
        if len(s) == 1:
            return 0
        total = 0
        arr = []
        pre = s[0]
        j = 0
        for i in range(1, len(s)):
            if s[i] != pre:
                arr.append(i-j)
                j = i
                pre = s[i]
        else:
            arr.append(i+1-j)
        for i in range(len(arr)-1):
            total += min(arr[i], arr[i+1])
        return total
最后修改日期: 2021年5月25日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。