Leetcode python (3/6)

Leetcode python