696. 计数二进制子串(Count Binary Substrings) - 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”字符。

详见代码

/**
 * 注意边界问题
 * <p>
 * 思路:受回文串启发，启用了中心(点)扩散的方式寻找01串
 * 步骤
 * 1.首先找到01的分界点
 * 2.找到分界点后进行前后同时查找
 */

class Solution_696 {
    public int isAns(String s, int index) {
        int i = index; //左边 index很重要
        int j = index + 1; //右边
        int sum = 1;
        while (i > 0 && j < s.length() - 1) { //左右扩散的时候: 左边与index位置字符相同,右边与index字符不相同
            if (s.charAt(--i) == s.charAt(index) && s.charAt(++j) != s.charAt(index)) {
                sum++;
            } else break;
        }
        return sum;
    }

    public int countBinarySubstrings(String s) {
        int sum = 0;
        for (int i = 0; i < s.length() - 1; i++) {
            if (s.charAt(i) != s.charAt(i + 1)) { //找到01分界线
                sum += isAns(s, i);
            }
        }
        return sum;
    }
}

public class _696 {
    public static void main(String[] args) {
        Solution_696 solution = new Solution_696();
        String s = "00110011";
        String s1 = "00110";
        System.out.println(solution.countBinarySubstrings(s));
        System.out.println(solution.countBinarySubstrings(s1));
    }
}