Problem Description
A = [0, 2, 1]Input 2:
A = [5, 3, 2, 5, 7]
[0, 0, 0]Output 2:
[0, 9, 16, 0, 0]
There is no subarray for the 0th element because it would always be the first element in each subarray. For the 1st element, there is only the subarray [0, 2, 1] which satisfies the first 2 conditions. But it doesn’t satisfy the 2nd condition as max(0, 2) = max(2). There is no subarray for 2nd element because it would always be the last element in each subarray.For Input 2:
There is no subarray for the 0th element because it would always be the first element in each subarray. The 2 subarrays that satisfy the conditions for the 1st element are [5, 3, 2, 5] and [5, 3, 2, 5, 7]. The sum of lengths is 4 + 5 = 9. A total of 4 subarrays satisfy the conditions for the 2nd element - [5, 3, 2, 5], [5, 3, 2, 5, 7], [3, 2, 5, 7] and [3, 2, 5]. Sum = 3 + 4 + 4 + 5 = 16. No subarray satisfies the conditions for the 3rd and 4th elements.
NOTE: You only need to implement the given function. Do not read input, instead use the arguments to the function. Do not print the output, instead return values as specified. Still have a question? Checkout Sample Codes for more details.