Array Reduce Transformation Leetcode Solution Python