Number of Substrings With Only 1s LeetCode solution in Python

1
# Runtime: 46 ms (Top 40.6%) | Memory: 14.50 MB (Top 56.2%)
2

3

4
class Solution(object):
5
    def numSub(self, s):
6
        res, currsum = 0, 0
7
        for digit in s:
8
            if digit == "0":
9
                currsum = 0
10
            else:
11
                currsum += 1
12
                res += currsum
13
        return res % (10**9 + 7)