仅含 1 的子串数

2020-07-28  本文已影响0人  WAI_f

题目:

给你一个二进制字符串 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<s.size();i++)
        {
            if(s[i]=='0')
            {
                cnt+=ones*(ones+1)/2;
                cnt%=1000000007;
                ones=0;
            }
            else
            {
                ones++;
            }
        }
        cnt+=ones*(ones+1)/2;
        cnt%=1000000007;
        
        return cnt;
    }
};
运行结果:

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

上一篇 下一篇

猜你喜欢

热点阅读