1. 程式人生 > >leetcode 696. 計數二進位制子串(python)

leetcode 696. 計數二進位制子串(python)

給定一個字串 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”字元。
class Solution(object):
    def countBinarySubstrings(self, s):
        """
        :type s: str
        :rtype: int
        """
        #利用map函式求出被切分開的0和1組合長度
        L = list(map(len, s.replace('01', '0 1').replace('10', '1 0').split(' ')))
        #因為限定了只能0和1挨著,所以只能相鄰的才可能組隊,用了zip函式
        res = sum(min(a,b) for a,b in zip(L, L[1:]) )
        return res

 知識點一:map()函式

map()是 Python 內建的高階函式,它接收一個函式 f 和一個 list,並通過把函式 f 依次作用在 list 的每個元素上,得到一個新的 list 並返回。

知識點二:zip函式

zip函式的原型為:zip([iterable, …])

引數iterable為可迭代的物件,並且可以有多個引數。該函式返回一個以元組為元素的列表,其中第 i 個元組包含每個引數序列的第 i 個元素。返回的列表長度被截斷為最短的引數序列的長度。只有一個序列引數時,它返回一個1元組的列表。沒有引數時,它返回一個空的列表。

 

方法二:

下邊的方法很好理解,但是無法AC(因為超出時間限制了)

class Solution:
    def countBinarySubstrings(self, s):
        """
        :type s: str
        :rtype: int
        """
        if(len(s)<2 or s.count('0')==len(s) or s.count('1')==len(s)):
            return 0
        s2='01'
        s3='10'
        out=0
        while(s.count(s2) or s.count(s3)):
            out+=s.count(s2)
            s2='0'+s2+'1'
            out+=s.count(s3)
            s3='1'+s3+'0'
        return out