leetcode python problems for data engineer