Problem
This is similar to Product of array except self
TC: O(n) ,SC:O(n)
# this is same as product of subarray except self class Solution: def minOperations(self, boxes: str) -> List[int]: res = [0]*len(boxes) # intialize balls and moves going from left to right balls,moves = 0,0 for i in range(len(boxes)): res[i] = balls+moves moves = balls+moves # moves should be updated before the no. of balls are updated, because this tell us the no of moves it took to mvoe all the ball to the current index balls+= int(boxes[i]) # once the moves are updated we an then updated no. of balls if the current index value is also 1 balls,moves = 0,0 # reinitialize them now going from right to left for i in reversed(range(len(boxes))): res[i]+= balls+moves # this time we just need to add the values intead of overriding as we have already calculated moves from left and from right we will just add to the existing moves present moves = balls+moves balls += int(boxes[i]) return res
Top comments (0)