Leetcode Basic Problems Python