LeetCode:696. 计数二进制子串

给定一个字符串 s,计算具有相同数量 0 和 1 的非空(连续)子字符串的数量,并且这些子字符串中的所有 0 和所有 1 都是连续的。

重复出现的子串要计算它们出现的次数。

示例 1:

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

请注意,一些重复出现的子串要计算它们出现的次数。

另外,“00110011”不是有效的子串,因为所有的0(和1)没有组合在一起。

 

这道题,纯粹的看能不能先找到规律,如果每找到规律真的很不好做。

我们可以将字符串 s 按照 0 和 1 的连续段分组,存在 counts 数组中,例如 s = 00111011,可以得到这样的 counts 数组:counts={2,3,1,2}。

这里 counts 数组中两个相邻的数一定代表的是两种不同的字符。假设 counts 数组中两个相邻的数字为 u 或者 v,它们对应着 u 个 0 和 v 个 1,或者 u 个 1 和 v 个 0。它们能组成的满足条件的子串数目为 min{u,v},即一对相邻的数字对答案的贡献。

当然在实际计算中,我们其实不需要将counts整个数组存起来再计算,而是应该利用动态规划只需要:当前值+上一次的值 即可求得答案。

class Solution {
    public int countBinarySubstrings(String s) {
       
        int len = s.length();
        int sum = 0;   //统计总数量
        int last = 0;  //上一次连续的0/1的数量

        int i = 0; //循环计数器
        while (i<len){
            char ch = s.charAt(i);
            int curr = 0;  //当前连续的0/1的数量
            while(i<len && s.charAt(i)==ch){
                curr++;
                i++;
            }

            sum += Math.min(last,curr);
            last = curr;
        }

        return sum;
    }
}

时间复杂度:O(n),空间复杂度:O(1)

 

posted @ 2021-04-28 15:33  蜗壳吃虾米  阅读(45)  评论(0)    收藏  举报