LeetCode - 139. 单词拆分

139. 单词拆分

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

说明：

• 拆分时可以重复使用字典中的单词。
• 你可以假设字典中没有重复的单词。

解题思路： 此题考查DP，Dynamic Programming，dp[len]表示s的前len个字符是否能拆分成字典中出现的多个单词，dp[len] = dp[len - a_0] || … || dp[len - a_n-1]

// DP,dp[len]表示s的前len个字符是否能拆分成字典中出现的多个单词，
// dp[len] = dp[len - a_0] || ... || dp[len - a_n-1]
// Time: O(n * mxlen), Space: O(n)
class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        int mxlen = 0;
        unordered_set<string> st(wordDict.begin(), wordDict.end());
        for (auto t : wordDict) {
            mxlen = max(mxlen, (int)t.size());
        }
        int n = s.size();
        vector<bool> dp(n + 1, false);
        dp[0] = true;
        for (int i = 1; i <= n; ++i) {
            for (int len = 1; len <= mxlen && len <= i; ++len) {
                if (st.count(s.substr(i - len, len))) {
                    dp[i] = dp[i] || dp[i - len];
                }
            }
        }
        return dp[n];
    }
};