leetcode696題計數二進制子串

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”字符。

題解：

方法一：

/**
 * @param {string} s
 * @return {number}
 */
//暴力枚舉（超出時間限制）
var countBinarySubstrings = function(s) {
    let len = s.length;
    let count = 0;
    let s1 = '';
    let s2 = '';
    for(let i = 1; i <= Math.floor(len / 2); i++) { 
      //最短匹配兩個字符，所以長度爲一半
        s1 += '0';
        s2 += '1';
        let res1 = s.match(new RegExp(s1 + s2, 'g')) || [];
        let res2 = s.match(new RegExp(s2 + s1, 'g')) || [];
        count += res1.length;
        count += res2.length;
    }
    return count;
};

如上面方法並不能在leetcode上成功的跑下來因爲res1和res2每次都對字符串s中的所有元素進行匹配，所以複雜度太高了導致運行都會超出時間限制。

方法二：

var countBinarySubstrings = function(s) { 
    let n = 0, pre = 0, curr = 1;
    for (let i = 0; i < s.length - 1; i++) { 
      //匹配兩個字符開始，所以s.length - 1
        if (s[i] == s[i+1]) {
            curr++;  //連續相等的長度
        } else {
            pre = curr; //後面連續符合的長度
            curr = 1;  
        }
        if (pre >= curr) n++; //符合一次加1
    }
    return n;
};

如上，比較大衆的解法，複雜度比一要低很多。