仅含 1 的子串数

题目:

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

示例:

输入:s = "0110111"
输出:9
解释:共有 9 个子字符串仅由 '1' 组成
"1" -> 5 次
"11" -> 3 次
"111" -> 1 次

解题方法:

  • 统计全1子串的长度ones;
  • 根据统计长度计算所有可能的全1子串的数量,计算公式:cnt=(ones+1)*ones/2

这个公式怎么来的呢,对于一个长度为n的全1字符串,有n个长度为1的全1子串,有n-1个长度为2的全1子串,...,有1个长度为n的全1子串,根据等差数列求和公式可以得到所有的可能数。

代码和结果:

class Solution {
public:
    int numSub(string s) {
        int i=0;
        long cnt=0;
        long ones=0;
        for(int i=0;i

运行结果:

原题链接:https://leetcode-cn.com/problems/number-of-substrings-with-only-1s/

你可能感兴趣的:(仅含 1 的子串数)