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

题目描述

给定一个字符串 s，计算具有相同数量0和1的非空(连续)子字符串的数量，并且这些子字符串中的所有0和所有1都是组合在一起的。重复出现的子串也要计算它们出现的次数。

输入: "00110011"
输出: 6
解释: 有6个子串具有相同数量的连续1和0：“0011”，“01”，“1100”，“10”，“0011” 和 “01”。

解题思路

一次遍历即可，关键是要理解操作步骤：分别记录“前一个连续字符数”和“当前连续字符数”即可，结果是我们看到的每个不同的 (prev, curr) 的 min(prev, curr) 之和。

举例：比如01，00111，0011，00011。

参考代码

class Solution {
public:
    int countBinarySubstrings(string str) {
        int prev = 0, curr = 1, res = 0;
        for(int i = 1; i < str.size(); i++){
            if(str[i] == str[i-1]){
                curr++;
            }else{
                res += min(curr, prev);
                prev = curr;
                curr = 1;
            }
        }
        
        res += min(curr, prev);
        return res;
    }
};