我用java刷 leetcode 1513. 仅含 1 的子串数

这里有leetcode题集分类整理！！！

题目难度：中等
题目描述：
给你一个二进制字符串 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

核心思路详解：
串为 “1” ： 组合个数为 1
串为 “11” ： 组合个数为 1+2
串为 “111”： 组合个数为 1+2+3
所以可以得出结论 长度 为n时 组合个数 等于 : n * (n + 1) / 2

class Solution {
    public int numSub(String s) {
        int len = s.length();
        long tempLen = 0;
        long res = 0;
        int mod = 1000000007;
        for (int i = 0 ; i < len ; i ++) {
            if (s.charAt(i) == '0') {
                res += tempLen * (tempLen + 1) >>> 1;
                res %= mod;
                tempLen = 0;
            } else {
                tempLen ++;
            }
        }
        res += tempLen * (tempLen + 1) >>> 1;
        res %= mod;
        return (int)res;
    }
}