Sort Colors Problem Leetcode Python Solutions