计数二进制子串 - 码农教程

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-binary-substrings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解决方案：

我们发现数据范围在5*1e4，暴力求解一定会tle。

题目要求获得相同数量的连0连1子串，因此我们可以使用一临时变量preCount存储之前那个连0/连1串的长度，curCount存储当前连0/连1串的长度。可以只遍历一次即可求解。

例如对于如下案例

"00111"

遍历到最后一个位置时，preCount = 2， curCount = 3。

此外我们发现对于每次能够组成满足条件的子串数目为min(preCount, curCount)

"00111"   
01 0011

代码如下：

class Solution {
    public int countBinarySubstrings(String s) {
        // init
        int preCount = 0;
        int curCount = 1;
        char cur = s.charAt(0);
        int ans = 0;
        for(int i = 1; i < s.length(); i++){
            if(s.charAt(i) == cur){
                curCount++;
            }else{
                ans += Math.min(preCount, curCount);
                preCount = curCount;
                curCount = 1;
                cur = s.charAt(i);
            }
        }
        ans += Math.min(preCount, curCount);
        return ans;
    }
}