-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy path130.py
More file actions
41 lines (34 loc) · 1.28 KB
/
130.py
File metadata and controls
41 lines (34 loc) · 1.28 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
class Solution:
def minZeroArray(self, nums: List[int], queries: List[List[int]]) -> int:
n = len(nums)
left, right = 0, len(queries)
# Zero array isn't formed after all queries are processed
if not self.can_form_zero_array(nums, queries, right):
return -1
# Binary Search
while left <= right:
middle = left + (right - left) // 2
if self.can_form_zero_array(nums, queries, middle):
right = middle - 1
else:
left = middle + 1
# Return earliest query that zero array can be formed
return left
def can_form_zero_array(
self, nums: List[int], queries: List[List[int]], k: int
) -> bool:
n = len(nums)
total_sum = 0
difference_array = [0] * (n + 1)
# Process query
for query_index in range(k):
start, end, val = queries[query_index]
# Process start and end of range
difference_array[start] += val
difference_array[end + 1] -= val
# Check if zero array can be formed
for num_index in range(n):
total_sum += difference_array[num_index]
if total_sum < nums[num_index]:
return False
return True