目标
给你一个二进制字符串 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
思路
统计全一子串的个数。
根据等差数列求和公式计算即可。
代码
/**
* @date 2025-11-16 22:17
*/
public class NumSub1513 {
public int numSub(String s) {
int mod = 1000000007;
int n = s.length();
int res = 0;
for (int i = 0; i < n; i++) {
if (s.charAt(i) == '0') {
continue;
}
int start = i;
while (i < n && s.charAt(i) == '1'){
i++;
}
int cnt = i - start;
res += ((cnt + 1L) * cnt / 2) % mod;
}
return res;
}
}
性能
