Leetcode python (4/6)

Leetcode python