LeetCode热题100 单词拆分

题目描述

给你一个字符串 s 和一个字符串列表 wordDict 作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true。

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

示例 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 <= wordDicti.length <= 20

s 和 wordDicti 仅由小写英文字母组成

wordDict 中的所有字符串 互不相同

思路

动态规划模板，使用哈希记录每个字符串，每次看第i个位置减去这些字符串后的位置是否为true即可，详细见代码。

代码

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        // 记录所有字符串
        unordered_set<string>ust;
        // 记录最长的字符串长度
        int maxs = 0; 
        for(const string &word : wordDict)
        {
            ust.insert(word);
            maxs = max(maxs, (int)word.length());
        }

        int n = s.length();
        vector<bool>dp(n, false);
        for(int i = 0; i < n; ++i)
        {
            int m = min(maxs, i + 1);
            for(int j = m; j; --j)
            {
                int k = i - j + 1;  
                if((k == 0 || dp[k - 1]) && ust.find(s.substr(k, j)) != ust.end())
                {
                    dp[i] = true;
                    break;
                }
            }
        }

        return dp[n - 1];
    }
};