leetcode.字符串.696计数二进制子串-java

1. 具体题目

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

2. 思路分析

看到题目，首先想到了leetcode20.有效的括号，将 '0' 和 1' 等效于括号，设置一个计数器。但是要遍历字符串2次，第一次遇到 '0' 计数器 + 1，遇到 '1' 计数器 - 1；第二次遇到 '1' 计数器 + 1，遇到 '0' 计数器 - 1。

3. 代码

 public int countBinarySubstrings(String s) {
         int count = 0;
         int res = 0;
         for(int i = 0; i < s.length(); i++){
             if(s.charAt(i) == '0') count++;
             if(s.charAt(i) == '1'){　　
                 if(count > 0){　　//若有可配对的"0"，结果 + 1
                     res++;
                     count--;
                 }
                 if(i + 1 < s.length() && s.charAt(i + 1) == '0'){　　//若无可配对的"1"，计数器置0
                     count = 0;
                 }
             }
         }
         count = 0;
         for(int i = 0; i < s.length(); i++){
             if(s.charAt(i) == '1') count++;
             if(s.charAt(i) == '0'){
                 if(count > 0){
                     res++;
                     count--;
                 }
                 if(i + 1 < s.length() && s.charAt(i + 1) == '1'){
                     count = 0;
                 }
             }
         }
         return res;
     }

4. 思路优化

实际上，字符串中若有 n 个 "0" 和 (n + m) 个 "1" 相邻，则说明可以产生 n 个重复字符串，所以只要统计数组中各连续子串中元素的数量即可，比如s=“11000111000000”，则可以得到一个统计数组groups = [2，3，4，6]，对该数组做遍历，每次得到 min(groups[i], groups[i+1])的值，将其全部相加可得最后结果。

5. 代码优化

 public int countBinarySubstrings(String s) {
         int[] groups = new int[s.length()];
         int t = 0;
         groups[0] = 1;
         for (int i = 1; i < s.length(); i++) {
             if (s.charAt(i-1) != s.charAt(i)) {
                 groups[++t] = 1;
             } else {
                 groups[t]++;
             }
         }

         int ans = 0;
         for (int i = 1; i <= t; i++) {
             ans += Math.min(groups[i-1], groups[i]);
         }
         return ans;
     }