仅含 1 的子串数

/*
 * 思路：
 * 使用 Java Stream 的方式进行解答。首先将字符串按 '0' 分割为子字符串。
 * 然后对于每一个子字符串计算其长度，使用 n * (n + 1) / 2 计算可能的
 * 子字符串数量。
 */
import java.util.Arrays;

public class Solution {
    public int numSub(String s) {
        int mod = 1000000007;
        return Arrays.stream(s.split("0"))
                     .mapToInt(str -> str.length() * (str.length() + 1) / 2)
                     .reduce(0, (a, b) -> (a + b) % mod);
    }
}