字符串题目：计数二进制子串

题目

标题和出处

标题：计数二进制子串

出处：696. 计数二进制子串

难度

4 级

题目描述

要求

给定一个二进制字符串 $\text{s}$，计算具有相同数量的 $\text{0}$ 和 $\text{1}$ 并且所有 $\text{0}$ 和所有 $\text{1}$ 都是连续的非空子字符串的数量。

重复出现的子串要计算它们出现的次数。

示例

示例 1：

输入：$\text{s = "00110011"}$
输出：$\text{6}$
解释：有 $\text{6}$ 个子串具有相同数量的连续 $\text{1}$ 和 $\text{0}$：$\text{"0011"}$，$\text{"01"}$，$\text{"1100"}$，$\text{"10"}$，$\text{"0011"}$ 和 $\text{"01"}$。
请注意，一些重复出现的子串要计算它们出现的次数。
另外，$\text{"00110011"}$ 不是有效的子串，因为所有的 $\text{0}$（和 $\text{1}$）没有组合在一起。

示例 2：

输入：$\text{s = "10101"}$
输出：$\text{4}$
解释：有 $\text{4}$ 个子串具有相同数量的连续 $\text{1}$ 和 $\text{0}$：$\text{"10"}$，$\text{"01"}$，$\text{"10"}$，$\text{"01"}$。

数据范围

• $\text{1}\le \text{s.length}\le {\text{10}}^{\text{5}}$
• $\text{s[i]}$ 是 $\text{‘0’}$ 或 $\text{‘1’}$

解法

思路和算法

可以将字符串 $s$ 分组，每一组只包含相同字符且相邻两组包含的字符不同。例如，$s=\text{“000011100"}$ 可以分成 $3$ 组：$\text{“0000"}$、$\text{“111"}$、$\text{“00"}$。

如果一个子串满足具有相同数量的 $0$ 和 $1$ 并且所有 $0$ 和所有 $1$ 都连续，则该子串满足以下两个条件：

• 该子串的长度为偶数；

• 该子串的前一半都是 $0$，后一半都是 $1$，或者前一半都是 $1$，后一半都是 $0$。

将字符串 $s$ 分组之后，满足上述条件的子串一定是跨越相邻的两个分组的某个子串。假设相邻的两个分组的长度分别是 $x$ 和 $y$，跨越这两个分组的子串包含 $z$ 个 $0$ 和 $z$ 个 $1$，则有 $1\le z\le \min (x,y)$，即跨越这两个分组的子串的数量是 $\min (x,y)$，它们的长度分别是从 $2$ 到 $2\times \min (x,y)$ 的全部偶数。

由此可以得到：跨越两个相邻的分组的子串数量是这两个分组的大小中的较小值。

基于上述结论，可以得到如下解法：遍历字符串 $s$ 并得到每个分组的大小，然后遍历每一对相邻的分组，将其中的较小的分组大小加到答案中，遍历结束后即可得到答案。

具体实现时，只需要遍历 $s$ 一次即可。遍历过程中判断相邻的两个字符是否相同，若相同则将当前分组的大小加 $1$，若不同则说明一个分组遍历结束，比较遍历结束的分组和更前面的分组的大小，将其中的较小的分组大小加到答案中，遍历 $s$ 结束时需要对最后一个分组再次计算，避免遗漏。

代码

class Solution {
    public int countBinarySubstrings(String s) {
        int count = 0, prev = 0, curr = 1;
        int length = s.length();
        for (int i = 1; i < length; i++) {
            if (s.charAt(i - 1) == s.charAt(i)) {
                curr++;
            } else {
                count += Math.min(prev, curr);
                prev = curr;
                curr = 1;
            }
        }
        count += Math.min(prev, curr);
        return count;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是字符串 $s$ 的长度。需要遍历字符串 $s$ 一次。

• 空间复杂度：$O(1)$。