力扣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

思路

对于一段连续的1，子串数满足等差数列求和
找规律：
// “1” 11
// “11” 32+1
// “111” 63+2+1
// “1111” 10 4+3+2+1
// “11111” 155+4+3+2+1
因此只要计算字符串有几个连续的1即可，在计算其等差和，等差数列求和公式
sum=n*（n+1）/2
有个小坑是计算等差数列时， number_ones * (number_ones + 1)会超int范围，因此需要用long接住在取模转int

代码

class Solution {public int numSub(String s) {char[] arr_s = s.toCharArray();int sum = 0;int number_ones = 0;for (char arr : arr_s) {if (arr == '1') {number_ones++;} else {if (number_ones != 0){int a = (int) (((double) ((long) number_ones * (number_ones + 1)) / 2)%(Math.pow(10,9)+7));sum+= (int) (a%(Math.pow(10,9)+7));}number_ones = 0;}}if (number_ones != 0)sum+= (int) (((double) (number_ones * (number_ones + 1)) /2)%(Math.pow(10,9)+7));return sum;}
}