Count Substrings With All 1s - Google Top Interview Questions


Problem Statement :


You are given a string s containing "1"s and "0"s. 
Return the number of substrings that contain only "1"s. Mod the result by 10 ** 9 + 7.

Constraints

0 ≤ n ≤ 100,000 where n is the length of s

Example 1

Input

s = "111001"

Output

7

Explanation

Here are all the substrings containing only "1"s:



"1"

"1"

"1"

"1"

"11"

"11"

"111"



Solution :



title-img




                        Solution in C++ :

int solve(string s) {
    int i = 0, ans = 0, mod = 1e9 + 7;
    for (int j = 0; j < s.length(); j++) {
        if (s[j] == '0')
            i = j + 1;
        else
            ans = (ans + j - i + 1) % mod;
    }
    return ans;
}
                    


                        Solution in Java :

import java.util.*;

class Solution {
    public int solve(String s) {
        int tempResult = 0;
        int result = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '1') {
                tempResult++;
            } else {
                while (tempResult > 0) {
                    result += tempResult;
                    tempResult--;
                }
            }
        }
        while (tempResult > 0) {
            result += tempResult;
            tempResult--;
        }

        return result;
    }
}
                    


                        Solution in Python : 
                            
class Solution:
    def solve(self, s):
        a = 0
        count = 0
        for i in range(len(s)):
            if s[i] == "0":
                a = 0
            else:
                a += 1
                count += a
        return count
                    


View More Similar Problems

Array-DS

An array is a type of data structure that stores elements of the same type in a contiguous block of memory. In an array, A, of size N, each memory location has some unique index, i (where 0<=i<N), that can be referenced as A[i] or Ai. Reverse an array of integers. Note: If you've already solved our C++ domain's Arrays Introduction challenge, you may want to skip this. Example: A=[1,2,3

View Solution →

2D Array-DS

Given a 6*6 2D Array, arr: 1 1 1 0 0 0 0 1 0 0 0 0 1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 An hourglass in A is a subset of values with indices falling in this pattern in arr's graphical representation: a b c d e f g There are 16 hourglasses in arr. An hourglass sum is the sum of an hourglass' values. Calculate the hourglass sum for every hourglass in arr, then print t

View Solution →

Dynamic Array

Create a list, seqList, of n empty sequences, where each sequence is indexed from 0 to n-1. The elements within each of the n sequences also use 0-indexing. Create an integer, lastAnswer, and initialize it to 0. There are 2 types of queries that can be performed on the list of sequences: 1. Query: 1 x y a. Find the sequence, seq, at index ((x xor lastAnswer)%n) in seqList.

View Solution →

Left Rotation

A left rotation operation on an array of size n shifts each of the array's elements 1 unit to the left. Given an integer, d, rotate the array that many steps left and return the result. Example: d=2 arr=[1,2,3,4,5] After 2 rotations, arr'=[3,4,5,1,2]. Function Description: Complete the rotateLeft function in the editor below. rotateLeft has the following parameters: 1. int d

View Solution →

Sparse Arrays

There is a collection of input strings and a collection of query strings. For each query string, determine how many times it occurs in the list of input strings. Return an array of the results. Example: strings=['ab', 'ab', 'abc'] queries=['ab', 'abc', 'bc'] There are instances of 'ab', 1 of 'abc' and 0 of 'bc'. For each query, add an element to the return array, results=[2,1,0]. Fun

View Solution →

Array Manipulation

Starting with a 1-indexed array of zeros and a list of operations, for each operation add a value to each of the array element between two given indices, inclusive. Once all operations have been performed, return the maximum value in the array. Example: n=10 queries=[[1,5,3], [4,8,7], [6,9,1]] Queries are interpreted as follows: a b k 1 5 3 4 8 7 6 9 1 Add the valu

View Solution →