leetcode 46 problem solution in python