Create Substrings_with_1.py

Author: Abhinavcode13

Math/Substrings_with_1.py

@@ -0,0 +1,29 @@
+'''Name : Abhinav kumar
+Github username : Abhinavcode13
+Repository name : data-structures-and-algorithms
+Problem : Number of Substrings With Only 1s in Python
+Issue Number : #518
+Problem statement : 
+
+Explanation of the below Python code :
+
+The implementation is quite similar to the C++ implementation. We iterate over the string s using a for loop and keep track of the count of consecutive 1's using the variable count. Whenever we encounter a '0', we calculate the number of substrings that can be formed using the formula n*(n+1)/2, add it to the final answer ans, and reset the value of count to 0. Finally, we calculate the number of substrings for the last substring of consecutive 1's, add it to the final answer ans, and return the result modulo 10^9 + 7. Note that we have used the integer division operator // to perform the division in Python.
+
+
+
+'''
+
+----------------------------------------------------------------------------------------------//Python code begins here--------------------------------------------------------------------------------------------------------------------------------------
+
+class Solution:
+    def numSub(self, s: str) -> int:
+        count = 0
+        ans = 0
+        for i in range(len(s)):
+            if s[i] == '1':
+                count += 1
+            else:
+                ans = (ans + (count*(count+1))//2) % (10**9 + 7)
+                count = 0
+        ans = (ans + (count*(count+1))//2) % (10**9 + 7)
+        return ans