Leetcode python (6/6)

Leetcode python