Word Pattern Leetcode Solution Python