[LeetCode] 696. 计数二进制子串

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

思路:

问题实际在问:对于每一对相邻的连续相同字符段——比如一段0和相邻的一段1——计算它们能形成多少有效子串。所以,对于长度为 len0 的0段和长度为 len1 的1段,可以形成 min(len0, len1) 个有效子串。那么第一步就是将字符串按连续相同字符分组,记录每组长度,第二步则是对于相邻的两段,累加 min(前一段, 后一段) 即可。

AC代码:

class Solution {
public:
    int countBinarySubstrings(string s) {
        if (s.empty()) return 0;
        
        vector<int> groups;
        int count = 1;
        
        for (int i = 1; i < s.size(); i++) {
            if (s[i] == s[i-1]) {
                count++;
            } else {
                groups.push_back(count);
                count = 1;
            }
        }
        groups.push_back(count);
        
        int result = 0;
        for (int i = 1; i < groups.size(); i++) {
            result += min(groups[i-1], groups[i]);
        }
        
        return result;
    }
};Code language: C++ (cpp)