Leetcode Problem 641 Solution