leetcode 1513. 仅含 1 的子串数

leetcode 1513. 仅含 1 的子串数（C语言）

题目链接：leetcode 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

先上代码：

int numSub(char * s){
    int i=0,j=0,max=0,len=strlen(s);
    int *dp=(long *)malloc(sizeof(long)*len);
    dp[0]=s[0]-'0';
    max=s[0]-'0';
    for(i=1;i<len;i++)
    {
        if(s[i]=='1')
        {
            dp[i]=(dp[i-1]+max+1)%1000000007;
            max=(max+1)%1000000007;
        }
        else
        {
            dp[i]=dp[i-1]%1000000007;
            max=0;
        }
    }
    return dp[len-1];
}

思路：动态规划
定义一个一维数组表示状态，数组表示的状态是当前第i个只含1的子串个数然后构建状态转移方程。
当s[i]=0时；很显然dp[i]=dp[i-1];
**当s[i]=1时；这里就要分析状态的变化：**当前为1时，首先增加了一个只包含本身一个1的新的字符串，但是我们还要统计这个1前面有多少个连续的1，因为前面有多少个连续的1，就意味着增加了多少个新的字符串，所以
选择用一个max来记录在当前1的位置前面的连续1的个数
所以可以的得出dp[i]=dp[i-1]+max+1;