Sort An Array Leetcode Solution Python