Leetcode-1513. 仅含 1 的子串数

链接

1513. 仅含 1 的子串数

题目

给你一个二进制字符串 s（仅由 '0' 和 '1' 组成的字符串）。
返回所有字符都为 1 的子字符串的数目。
由于答案可能很大，请你将它对 10^9 + 7 取模后返回。

示例

示例 1：
输入：s = "0110111"
输出：9
解释：共有 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

思路

遍历字符串，以0为分隔符，统计每段连续1的个数，连续1的长度决定子集个数，公式为n(1+n)/2,将每一段的子集个数加起来即可。

C++ Code

class Solution {
public:
    int numSub(string s) {
        //以0位分割符 统计连续1的长度
        long int count=0;
        int i=0;
        while(i<s.size())
        {
            while(i<s.size() && s[i]=='0') i++;
            long int N=0;
            while(i<s.size() && s[i]=='1')
            {
                 i++; N++;
            }
            count+=(N+N*N)/2;
        }
        return count%(1000000007);
    }
};