Make Array Beautiful Leetcode Python