leetcode 5461. 仅含 1 的子串数（C++）

给你一个二进制字符串 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

C++

class Solution {
public:
    int numSub(string s) 
    {
        int mod=1000000007;
        long res=0;
        int start=0;
        int n=s.size();
        map<int,int> tmp;
        int num=0;
        for(int i=0;i<n;i++)
        {
            if('1'==s[i])
            {
                num++;
            }
            else
            {
                if(num)
                {
                    tmp[num]++;
                    num=0;
                }
            }
        }
        if(num)
        {
            tmp[num]++;
        }
        for(auto it:tmp)
        {
            res+=(long)it.first*(long)(it.first+1)/2*it.second%mod;
        }
        return res%mod;
    }
};