字符串 - 统计二进制字符串中连续 1 和连续 0 数量相同的子字符串个数

1.题目

LeetCode: 696. 计数二进制子串

【easy】

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

2.解题

方法一：拓展法

遍历字符串，找到两个相邻的不同的字符进行拓展。

java：

class Solution {
    int res;
    public int countBinarySubstrings(String s) {
        res = 0;
        for (int i = 0; i < s.length() - 1; i++) {
            if (s.charAt(i) != s.charAt(i + 1)) {
                res++;
                extend(s, i, i + 1);
            }
        }
        return res;
    }

    public void extend(String s, int start, int end) {
        while (start-1 >= 0 && end+1 < s.length() && s.charAt(start) == s.charAt(start - 1) && s.charAt(end) == s.charAt(end + 1)) {
            res++;
            start--;
            end++;
        }
    }
}

*方法二：一次遍历法

定义一个变量pre_len记录上一次连续相同字符的长度，cur_len记录当前连续相同字符的长度，然后进行比较，如果pre_len >= cur_len，则证明含有符合要求的子字符串。

java：

class Solution {
    public int countBinarySubstrings(String s) {
        int res = 0, cur_len = 1, pre_len = 0;
        for (int i = 1; i < s.length(); i++) {
            if (s.charAt(i) == s.charAt(i - 1)) {
                cur_len++;
            } else {
                pre_len = cur_len;
                cur_len = 1;
            }
            if (pre_len >= cur_len) res++;
        }
        return res;
    }
}

时间复杂度：O(n)

空间复杂度：O(1)