Leetcode Solution Python Tree