LeetCode——696. 计数二进制子串

题目描述：

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

提示：

• s.length 在1到50,000之间。
• s 只包含“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。

代码如下:

class Solution {
    public int countBinarySubstrings(String s) {
        int n = s.length();
        char[] ch = s.toCharArray();
        int a = 0, b = 0;
        ArrayList<Integer> arr = new ArrayList<>();
        for (int i = 0; i < n - 1; i++) {
            if (ch[i] == '0') {
                a++;
            }
            if (ch[i] == '1') {
                b++;
            }
            if ((ch[i] == '0' && ch[i + 1] == '1') || (ch[i] == '1' && ch[i + 1] == '0')) {
                if (a != 0) {
                    arr.add(a);
                }
                if (b != 0) {
                    arr.add(b);
                }
                a = 0;
                b = 0;
            }
        }
        if (ch[n - 1] == '0') {
            a++;
            arr.add(a);
        }
        if (ch[n - 1] == '1') {
            b++;
            arr.add(b);
        }
        int sum = 0;
        for (int i = 0; i < arr.size() - 1; i++) {
            sum += Math.min(arr.get(i), arr.get(i + 1));
        }
        return sum;
    }
}

class Solution {
    public int countBinarySubstrings(String s) {
        List<Integer> counts = new ArrayList<Integer>();
        int ptr = 0, n = s.length();
        while (ptr < n) {
            char c = s.charAt(ptr);
            int count = 0;
            while (ptr < n && s.charAt(ptr) == c) {
                ++ptr;
                ++count;
            }
            counts.add(count);
        }
        int ans = 0;
        for (int i = 1; i < counts.size(); ++i) {
            ans += Math.min(counts.get(i), counts.get(i - 1));
        }
        return ans;
    }
}

执行结果：