Python | Product of Prefix in list
Nowdays, especially in the field of competitive programming, the utility of computing prefix product is quite popular and features in many problems. Hence, having a one liner solution to it would possess a great help. Let’s discuss certain way in which this problem can be solved.
Method : Using list comprehension + list slicing This problem can be solved using the combination of above two functions in which we use list comprehension to extend logic to each element and then later compute the product, slicing is used to get product till the particular index.
The original list : [3, 4, 1, 7, 9, 1] The prefix prod list is : [3, 12, 12, 84, 756, 756]
Method : Using numpy
Note: Install numpy module using command “pip install numpy”
This problem can be solved by using numpy library which has the cumprod() function that can be used to calculate the cumulative product of elements in the list
The original list : [3, 4, 1, 7, 9, 1]
The prefix prod list is : [ 3 12 12 84 756 756]
The time complexity of this approach is O(n) and the auxiliary space is O(n)