Advertisement
Iam_Sandeep

Sum of minimum and maximum elements of all subarrays of size k.

Feb 23rd, 2022
677
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.82 KB | None | 0 0
  1. link="https://www.codingninjas.com/codestudio/problems/sum-of-minimum-and-maximum-elements-of-all-subarrays-of-size-k_1171047?leftPanelTab=1"
  2. from collections import deque
  3. def sumOfMaxAndMin(nums, n, k):
  4.     # write your code here
  5.     # return an integer denoting the answer
  6.     q=deque()
  7.     ans=0
  8.     for i,val in enumerate(nums):
  9.         while q and nums[q[-1]]<val:
  10.             q.pop()
  11.         q.append(i)
  12.         if q[0]==i-k:
  13.             q.popleft()
  14.         if i>=k-1:
  15.             ans+=nums[q[0]]
  16.     q=deque()
  17.     for i,val in enumerate(nums):
  18.         while q and nums[q[-1]]>val:
  19.             q.pop()
  20.         q.append(i)
  21.         if q[0]==i-k:
  22.             q.popleft()
  23.         if i>=k-1:
  24.             ans+=nums[q[0]]
  25.    
  26.     return ans
  27.        
  28.        
  29.        
  30.        
  31.            
  32.        
  33.            
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement