leetcode 77 problem solution in python