Sort Colors Leetcode Solution In Python Iteration