Leetcode Array Sort Python Function Arguments