leetcode 696. 计数二进制子串

leetcode 696. 计数二进制子串

题目详情

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

我的代码

class Solution {
public:
    int countBinarySubstrings(string s) {
        int zero = 0, one = 0;
        int n = s.size();
        int res = 0;
        int i = 0;
        while (i < n) {
            int loc = i;
            if (s[i] == '0') {
                while ((loc < n) && (s[loc] == '0')) {
                    ++zero, ++loc;
                }
                i = loc;
                res += min(one, zero);
                one = 0;
            } else {
                while ((loc < n) && (s[loc] == '1')) {
                    ++one, ++loc;
                }
                i = loc;
                res += min(one, zero);
                zero = 0;
            }
        }
        return res;
    }
};

我的成绩

执行结果：通过
执行用时：40 ms, 在所有 C++ 提交中击败了71.71%的用户
内存消耗：10.5 MB, 在所有 C++ 提交中击败了72.82%的用户

一些想法

本道题是一道简单题，我们要做的就是尽可能的减少重复计算的次数，只需进行一次遍历。

执行用时为 24 ms 的范例

class Solution {
public:
    int countBinarySubstrings(string s) {
        int n = s.size();
        int dp[n];
        dp[0] = 0;
        int zero = 0;
        int one = 0;
        char now = s[0];
        if(now == '0') zero++;
        else one++;
        for(int i = 1; i < n; i++){
            if(s[i] == now){
                if(now == '0'){
                    zero++;
                }
                else one++;
            }
            else{
                if(now =='0'){
                    now  = '1';
                    one = 1;
                }
                else{
                    now ='0';
                    zero = 1;
                }
            }
            if(now == '0' && zero <= one) dp[i] = dp[i-1] + 1;
            else if(now =='1' && one <= zero) dp[i] = dp[i-1] + 1;
            else dp[i] = dp[i-1];
        }
        return dp[n-1];
    }
};

思考

参见官方解答