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

代码参考:

package main

import "fmt"

func main() {
    fmt.Println(countBinarySubstrings("00110011"))
}

// 理解题意:相同数量 0 和 1 的子串
// 统计遍历考虑处理第一个元素后,从第二个开始向前遍历
// 相比于从第一个遍历到倒数第二个要简洁易处理
func countBinarySubstrings(s string) int {
    if len(s) <= 0 {
        return 0
    }

    zeroCount, oneCount := 0, 0
    if s[0] == '0' {
        zeroCount++
    } else {
        oneCount++
    }

    counts := 0
    for i := 1; i < len(s); i++ {
        switch s[i] {
        case '0':
            if s[i-1] == '0' {
                zeroCount++
            } else {
                zeroCount = 1
            }
            if oneCount >= zeroCount { // 对称
                counts++
            }
        case '1':
            if s[i-1] == '1' {
                oneCount++
            } else {
                oneCount = 1
            }
            if zeroCount >= oneCount {
                counts++
            }
        }
    }
    return counts
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng