Sort Colors Leetcode Solution In Python