1. 问题描述:
给你一个二进制字符串 s(仅由 '0' 和 '1' 组成的字符串)。
返回所有字符都为 1 的子字符串的数目。
由于答案可能很大,请你将它对 10^9 + 7 取模后返回。
示例 1:
解释:共有 9 个子字符串仅由 '1' 组成
"1" -> 5 次
"11" -> 3 次
"111" -> 1 次
示例 2:
输入:s = "101"
输出:2
解释:子字符串 "1" 在 s 中共出现 2 次
示例 3:
输入:s = "111111"
输出:21
解释:每个子字符串都仅由 '1' 组成
示例 4:
输入:s = "000"
输出:0
提示:
s[i] == '0'
或s[i] == '1'
1 <= s.length <= 10^5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-substrings-with-only-1s
2. 思路分析:
① 首先是需要分析出题目中的连续的1的数目子串是怎么样计算出来的,只有弄懂了这个问题才能够接下来计算相应的数目,比如111,由3个1,2个11,1个111组成,总共为6,1111为4个1,3个11,2个111,1个1111组成,为10很明显这个连续的1是一个等差序列
② 弄清楚了这个计算方式之后,那么我们需要在遍历的过程中计算出连续的1的数目是多少,这个其实也很简单,只需要使用一个变量对连续的1进行累加,当发现是0的时候对连续的1进行计数即可,最后还需要注意可能最末尾还存在连续的1所以还需要进行计算
③ 因为结果可能很大,所以在计算出等差序列的结果之后需要先对结果进行取余,防止存入之后溢出了,感觉弄懂了整个数目的计算过程还是比较简单的,抓住题目的关键即可
3. 代码如下:
class Solution:
def numSub(self, s: str) -> int:
# 两个乘号表示乘方的意思
mod = 10 ** 9 + 7
# 用来计算连续的1的能够形成的数目, 比如111形成1 + 2 + 3的数目其他的也是一样的
def cal(n):
return (1 + n) * n // 2 % mod
count, res = 0, 0 # 使用一个变量就可以累加计算出连续的1的数目
for i in range(len(s)):
if s[i] == '1':
count += 1
else:
# 不是1那么计算累加的1的数目 说明连续的1断掉了
res = (res + cal(count)) % mod
# 将计数恢复为0
count = 0
# 需要计算最后某位可能的1
if count > 0: res = (res + cal(count)) % mod
return res