[C++] LeetCode 139. 单词拆分

题目

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，确定 s 是否可以被空格分割为一个或多个在字典里出现的单词。你可以假设字典中无重复的单词。
例如，给出
s = "leetcode"，
dict = ["leet", "code"]。

返回 true 因为"leetcode" 可以被切分成 "leet code"。

更新 (2017/1/4):
wordDict 参数已被更改为字符串列表（而不是一组字符串）。请重新加载代码定义以获取最新更改。

题解

这道题可以先用一个unordered_set保存wordDict，方便后续查找。然后用动态规划，数组dp[i]表示s的前i个字符构成的子串能否由wordDict表示。

代码

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        int n=s.size();
        vector<bool> dp(n+1,false);
        unordered_set<string> ss;
        for(auto &s:wordDict) ss.insert(s);
        dp[0]=true;
        for(int i=1;i<=n;i++){
            for(int j=i;j>0;j--){
                if(dp[j-1]==false) continue;
                if(ss.find(s.substr(j-1,i-j+1))!=ss.end()){
                    dp[i]=true;
                    break;
                }
            }
        }
        return dp[n];
    }
};