leetcode hot100 之 单词拆分

题目

给定一个字符串，和一个字典。请判断该字符串能否有字典中的词拼接产生。
字典中的词可以重复使用，也可以不使用。

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

原题链接：https://leetcode.cn/problems/word-break/

思路

动态规划。设 dp[i] 表示以 i 为结尾的子字符串能否由字典中的词产生。那么
dp[i] = dp[j] & check(j, i) (j < i)
也就是，如果 dp[j] 能由字典产生， s[j, i] 刚好是字典中的词，则 dp[i] 也可以由字典产生。
那么这里就有两层循环：
for i = 0: n
for j = 0 : i
一旦有满足 dp[i] = true 时，则提前break。

同时，初始化 dp[0] = true，即空字符串能由字典中的词产生。

• 复杂度分析
  • 时间复杂度 O(n^2)。两层循环。实际有剪枝。最坏情况 O(n^2)
  • 空间复杂度 O(n)。

代码

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        unordered_set<string> wordSet;
        for (string word : wordDict) {
            wordSet.insert(word);
        }
        vector<bool> dp(s.size() + 1);
        dp[0] = true;
        for (int i = 1; i <= s.size(); i++) {
            for (int j = 0; j < i; j++) {
                dp[i] = dp[j] && wordSet.count(s.substr(j, i - j));
                if (dp[i]) {
                    break;
                }
            }
        }
        return dp[s.size()];
    }
};