leetcode data structures python