Sort Colors Problem Leetcode Python Interview