题目描述

给你一个字符串 s 和一个字符串列表 wordDict 作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true。
注意：不要求字典中出现的单词全部都使用，并且字典中的单词可以重复使用。

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

题目分析

动态规划。
设 ans[i] 表示是否能拼接出 s 中长度为 i 的前缀，则 ans[i] = ans[i] || ans[i - 字典中的单词长度]。
枚举字典单词或后缀长度都可以，只要为 true 就退出，然后计算下一个位置。

Java

public boolean wordBreak(String s, List<String> wordDict) {
    Set<String> set = new HashSet<>(wordDict);
    boolean[] ans = new boolean[s.length() + 1];
    ans[0] = true;
    // 枚举 s 长度 i
    for (int i = 1; i <= s.length(); i++) {
        // 枚举右侧长度 l
        for (int l = 1; l <= i; l++) {
            if (set.contains(s.substring(i - l, i))) {
                ans[i] = ans[i] || ans[i - l];
            }
            if (ans[i]) {
                break;
            }
        }
    }
    return ans[s.length()];
}

Kotlin

fun wordBreak(s: String, wordDict: List<String>): Boolean {
    val ans = BooleanArray(s.length + 1)
    val set = wordDict.toSet()
    ans[0] = true
    for (i in 1..s.length) {
        for (j in 1..i) {
            if (set.contains(s.substring(i - j, i))) {
                ans[i] = ans[i] || ans[i - j]
            }
            if (ans[i]) {
                break;
            }
        }
    }
    return ans.last()
}