题目

给你一个字符串 s 和一个字符串列表 wordDict 作为字典。请你判断是否可以利用字典中出现的单词拼接出 s 。

**注意：**不要求字典中出现的单词全部都使用，并且字典中的单词可以重复使用。

示例 1：

输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以由 "leet" 和 "code" 拼接成。

示例 2：

输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以由 "apple" "pen" "apple" 拼接成。
     注意，你可以重复使用字典中的单词。

示例 3：

输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
输出: false

提示：

• 1 <= s.length <= 300
• 1 <= wordDict.length <= 1000
• 1 <= wordDict[i].length <= 20
• s 和 wordDict[i] 仅由小写英文字母组成
• wordDict 中的所有字符串 互不相同

题解

static class Trie {
    Trie[] children = new Trie[26];
    boolean isLeaf;

    Trie insert(char c) {
        Trie trie = this.get(c);
        if (trie != null) {
            return trie;
        }

        trie = new Trie();
        this.children[c - 'a'] = trie;

        return trie;
    }

    Trie get(char c) {
        return this.children[c - 'a'];
    }

    void setLeaf() {
        this.isLeaf = true;
    }
}

public boolean wordBreak(String s, List<String> wordDict) {
    Trie root = new Trie();
    // 构建字典树
    for (String word : wordDict) {
        Trie node = root;
        for (char c : word.toCharArray()) {
            node = node.insert(c);
        }
        node.setLeaf();
    }

    int len = s.length();
    // 记录访问过的位置
    Set<Integer> visited = new HashSet<>();
    Function<Integer, Boolean> backtrack = new Function<Integer, Boolean>() {
        @Override
        public Boolean apply(Integer index) {
            if (index == len) {
                return true;
            }

            Trie node = root;
            for (int i = index; i < len; i++) {
                node = node.get(s.charAt(i));
                if (node == null) {
                    break;
                }

                // 截断单词
                if (node.isLeaf && !visited.contains(i + 1) && this.apply(i + 1)) {
                    return true;
                }

                // 不截断继续匹配
            }

            visited.add(index);
            return false;
        }
    };

    return backtrack.apply(0);
}