【Lintcode】1870. number of substrings with all zeroes

题目地址：

https://www.lintcode.com/problem/number-of-substrings-with-all-zeroes/description

给定一个$0-1$字符串$s$，求其只含$0$的子串个数。

考虑一个长度为$l$的只含$0$的字符串，则其所有子串都满足条件。考虑其有多少个子串。首先，其长度为$1$的子串有$l$个，其长度大于$1$的子串的个数，等价于在$1,...,l$中选取两个数的方式数，所以是$\left(\genfrac{}{}{0ex}{}{l}{2}\right)=\frac{l(l-1)}{2}$；所以总共有$l+\frac{l(l-1)}{2}=\frac{l(l+1)}{2}$个子串。

接着只需要将$s$中的最长$0$子串都截出来累加起来即可。代码如下：

public class Solution {
    /**
     * @param str: the string
     * @return: the number of substrings
     */
    public int stringCount(String str) {
        // Write your code here.
        int res = 0;
        for (int i = 0; i < str.length(); i++) {
            char ch = str.charAt(i);
            // 找到0了，则将以它开头的最长0子串截出来
            if (ch == '0') {
                int j = i + 1;
                while (j < str.length() && str.charAt(j) == '0') {
                    j++;
                }
                
                // s[i, ..., j - 1]就是一个0子串，其子串个数是l(l + 1)/2
                res += (j - i) * (j - i + 1) / 2;
				// 将i移到0后面去
                i = j;
            }
        }
        
        return res;
    }
}

时间复杂度$O(n)$，空间$O(1)$。