python solution for leetcode - View it on GitHub