題目:
給你一個(gè)二進(jìn)制字符串 s(僅由 '0' 和 '1' 組成的字符串)。
返回所有字符都為 1 的子字符串的數(shù)目如绸。
由于答案可能很大,請(qǐng)你將它對(duì) 10^9 + 7 取模后返回昼激。
示例:
輸入:s = "0110111"
輸出:9
解釋:共有 9 個(gè)子字符串僅由 '1' 組成
"1" -> 5 次
"11" -> 3 次
"111" -> 1 次
解題方法:
- 統(tǒng)計(jì)全1子串的長(zhǎng)度ones拖陆;
- 根據(jù)統(tǒng)計(jì)長(zhǎng)度計(jì)算所有可能的全1子串的數(shù)量,計(jì)算公式:
cnt=(ones+1)*ones/2
教届;
這個(gè)公式怎么來(lái)的呢响鹃,對(duì)于一個(gè)長(zhǎng)度為n的全1字符串,有n個(gè)長(zhǎng)度為1的全1子串案训,有n-1個(gè)長(zhǎng)度為2的全1子串买置,...,有1個(gè)長(zhǎng)度為n的全1子串强霎,根據(jù)等差數(shù)列求和公式可以得到所有的可能數(shù)忿项。
代碼和結(jié)果:
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;
}
};
運(yùn)行結(jié)果:原題鏈接:https://leetcode-cn.com/problems/number-of-substrings-with-only-1s/