139. 单词拆分

题目描述：给定一个非空字符串 s 和一个包含非空单词的列表 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。
说明：
拆分时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。
解题思路：trie+动态规划，前缀树用来快速查找当前字符串是否在字典中，dp数组用来记录哪些子串的位置可以用字典中单词构成，代码如下：

class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        class Trie(object):
            def __init__(self):
                self.root = {}
                self.end_word = -1
            
            def insert(self, s):
                n = len(s)
                i = 0
                curnode = self.root
                for c in s:
                    if c not in curnode:
                        curnode[c] = {}
                    curnode = curnode[c]
                curnode[self.end_word] = True
            
        trie = Trie()
        for c in wordDict:
            trie.insert(c)
        n = len(s)
        dp = [False] * (1 + n)
        dp[0] = True
        for i in range(1, n+1):
            if dp[i-1]:
                j = i - 1
                root = trie.root
                while(j < n and s[j] in root):
                    root = root[s[j]]
                    if -1 in root:
                        dp[j+1] = True
                    j += 1
        return dp[-1]