<p>You are given a binary string <code>s</code>.</p>

<p>Return the number of <span data-keyword="substring-nonempty">substrings</span> with <strong>dominant</strong> ones.</p>

<p>A string has <strong>dominant</strong> ones if the number of ones in the string is <strong>greater than or equal to</strong> the <strong>square</strong> of the number of zeros in the string.</p>

<p>&nbsp;</p>
<p><strong class="example">Example 1:</strong></p>

<div class="example-block">
<p><strong>Input:</strong> <span class="example-io">s = &quot;00011&quot;</span></p>

<p><strong>Output:</strong> <span class="example-io">5</span></p>

<p><strong>Explanation:</strong></p>

<p>The substrings with dominant ones are shown in the table below.</p>
</div>

<table>
	<thead>
		<tr>
			<th>i</th>
			<th>j</th>
			<th>s[i..j]</th>
			<th>Number of Zeros</th>
			<th>Number of Ones</th>
		</tr>
	</thead>
	<tbody>
		<tr>
			<td>3</td>
			<td>3</td>
			<td>1</td>
			<td>0</td>
			<td>1</td>
		</tr>
		<tr>
			<td>4</td>
			<td>4</td>
			<td>1</td>
			<td>0</td>
			<td>1</td>
		</tr>
		<tr>
			<td>2</td>
			<td>3</td>
			<td>01</td>
			<td>1</td>
			<td>1</td>
		</tr>
		<tr>
			<td>3</td>
			<td>4</td>
			<td>11</td>
			<td>0</td>
			<td>2</td>
		</tr>
		<tr>
			<td>2</td>
			<td>4</td>
			<td>011</td>
			<td>1</td>
			<td>2</td>
		</tr>
	</tbody>
</table>

<p><strong class="example">Example 2:</strong></p>

<div class="example-block">
<p><strong>Input:</strong> <span class="example-io">s = &quot;101101&quot;</span></p>

<p><strong>Output:</strong> <span class="example-io">16</span></p>

<p><strong>Explanation:</strong></p>

<p>The substrings with <strong>non-dominant</strong> ones are shown in the table below.</p>

<p>Since there are 21 substrings total and 5 of them have non-dominant ones, it follows that there are 16 substrings with dominant ones.</p>
</div>

<table>
	<thead>
		<tr>
			<th>i</th>
			<th>j</th>
			<th>s[i..j]</th>
			<th>Number of Zeros</th>
			<th>Number of Ones</th>
		</tr>
	</thead>
	<tbody>
		<tr>
			<td>1</td>
			<td>1</td>
			<td>0</td>
			<td>1</td>
			<td>0</td>
		</tr>
		<tr>
			<td>4</td>
			<td>4</td>
			<td>0</td>
			<td>1</td>
			<td>0</td>
		</tr>
		<tr>
			<td>1</td>
			<td>4</td>
			<td>0110</td>
			<td>2</td>
			<td>2</td>
		</tr>
		<tr>
			<td>0</td>
			<td>4</td>
			<td>10110</td>
			<td>2</td>
			<td>3</td>
		</tr>
		<tr>
			<td>1</td>
			<td>5</td>
			<td>01101</td>
			<td>2</td>
			<td>3</td>
		</tr>
	</tbody>
</table>

<p>&nbsp;</p>
<p><strong>Constraints:</strong></p>

<ul>
	<li><code>1 &lt;= s.length &lt;= 4 * 10<sup>4</sup></code></li>
	<li><code>s</code> consists only of characters <code>&#39;0&#39;</code> and <code>&#39;1&#39;</code>.</li>
</ul>

class Solution:
    def numberOfSubstrings(self, s: str) -> int:
        zero_locs = [i for i, num in enumerate(s) if num == '0']
        total_zeros = len(zero_locs)
        total_ones = len(s) - total_zeros
        answer = sum([(b - a - 1) * (b - a) // 2 for a, b in pairwise([-1] + zero_locs + [len(s)])])
        for num_zeros in range(1, total_zeros+1):
            ones_needed = num_zeros**2
            if ones_needed > total_ones:
                break
            end = len(zero_locs) - num_zeros + 1
            for start in range(end):
                prev = -1 if start == 0 else zero_locs[start-1]
                next = len(s) if start+num_zeros >= len(zero_locs) else zero_locs[start+num_zeros]

                sublocs = zero_locs[start:start+num_zeros]
                middle_ones_count = 0 if len(sublocs) == 1 else sublocs[-1] - sublocs[0] - len(sublocs) + 1
                left_ones_count = sublocs[0] - prev - 1
                right_ones_count = next - sublocs[-1] - 1

                current_needed = max(0, ones_needed - middle_ones_count)
                substring = s[sublocs[0]:sublocs[-1]+1]
                to_add = 0
                if left_ones_count + right_ones_count < current_needed:
                    to_add = 0
                else:
                    left = min(current_needed, left_ones_count)
                    right = current_needed - left
                    to_add += (left_ones_count - left + 1) * (right_ones_count - right + 1)
                    left -= 1
                    right += 1
                    while left >= 0 and right <= right_ones_count:
                        to_add += (right_ones_count - right + 1)
                        left -= 1
                        right += 1
                answer += to_add

        return answer


19300