【算法】计数二进制子串

zjk 发布于 2024-04-07 116 次阅读


题目

原题:力扣 - 计数二进制子串

给定一个字符串 s,统计并返回具有相同数量 0 和 1 的非空(连续)子字符串的数量,并且这些子字符串中的所有 0 和所有 1 都是成组连续的。重复出现(不同位置)的子串也要统计它们出现的次数。

示例 1:

输入:s = "00110011" 输出:6 解释:6 个子串满足具有相同数量的连续 1 和 0 :"0011"、"01"、"1100"、"10"、"0011" 和 "01" 。 注意,一些重复出现的子串(不同位置)要统计它们出现的次数。 另外,"00110011" 不是有效的子串,因为所有的 0(还有 1 )没有组合在一起。

示例 2:

输入:s = "10101" 输出:4 解释:有 4 个子串:"10"、"01"、"10"、"01" ,具有相同数量的连续 1 和 0 。

提示:

  • 1 <= s.length <= 105
  • s[i] 为 '0' 或 '1'

解题

class Solution:
    def countBinarySubstrings(self, s: str) -> int:
        pre, cur, res, prec = 0, 1, 0, s[0]
        for c in s[1:]:
            if c != prec: pre, cur = cur, 1
            else: cur += 1
            print(pre, cur, res, prec)
            if cur <= pre: res += 1
            prec = c
        return res

这段代码是用来计算具有相同数量的连续 0 和 1 的子串数目。它通过遍历输入的字符串 s 来实现这一目标。

首先,它初始化了四个变量:precurres 和 precpre 和 cur 用于跟踪当前和先前数字连续出现的次数。res 用于存储符合条件的子串数目,而 prec 用于存储前一个字符。

接下来,它遍历字符串 s 中的字符。对于每个字符 c,它执行以下操作:

  1. 如果当前字符 c 不等于前一个字符 prec,说明遇到了新的数字,此时将 pre 的值更新为 curcur 的值更新为 1。
  2. 如果当前字符 c 等于前一个字符 prec,则将 cur 的值加一,表示当前数字连续出现的次数。
  3. 然后,它检查当前的连续数量 cur 是否小于或等于先前的连续数量 pre,如果是,就增加计数器 res

最后,返回 res 作为结果,表示具有相同数量的连续 0 和 1 的子串数目。