140. 单词拆分 II--leecode

题目

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，在字符串中增加空格来构建一个句子，使得句子中所有的单词都在词典中。返回所有这些可能的句子。
说明：
分隔时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。

题目链接

解题思路

这道题明显是个搜索的题目。
但问题在于如何进行搜索。每个位置可能变成除自身以外的25个字母。直接搜会很暴力。
所以我们想了一种利用dp+字典树的方法来先将单词进行划分。
若dp[i]=j，表示字符串从下标i开始，长度为j-i的字符串存在在题目给定的字典里面。
如何查找呢？最好的方法就是将字典里面的单词建立一个字典树，然后在树上查找就好了。
在建立dp数组之后，需要从末尾开始DFS搜索结果就好了。

代码如下：

//字典树的模板
class Tire{
public:
    bool isLeaf;
    Tire* next[26];
    Tire(){
        isLeaf = false;
        memset(next,0,sizeof(next));
    }
    void Insert(string s){
        Tire *node = this;
        for(char c : s){
            if(node->next[c - 'a'] == NULL){
                node->next[c - 'a'] = new Tire();
            }
            node = node->next[c - 'a'];
        }
        node->isLeaf = true;
    }
};

class Solution {
public:
    void DFS(string s,vector<string>& ans,vector<string>& tWord,vector<vector<int>>dp,int d){
        if(d == 0){
            string str ="";
            for(auto it = tWord.rbegin();it != tWord.rend();it++){
                str += *it+" ";
            }
            str.pop_back();//最后末尾多余一个" "
            ans.push_back(str);
            return;
        }
        for(int i : dp[d]){
            tWord.push_back(s.substr(i,d - i));
            DFS(s,ans,tWord,dp,i);
            tWord.pop_back();
        }
    }
    vector<string> wordBreak(string s, vector<string>& wordDict){
        Tire* root = new Tire();
        for(string word : wordDict)    root->Insert(word);
        int l = s.size();
        vector<vector<int> > dp(l + 1);
        dp[0].push_back(-1);   
        for(int i = 0;i < l;i++){
            if(dp[i].empty()) continue;
            int j = i;
            Tire* cur = root;
            while(j < l && cur->next[s[j] - 'a']){
                cur = cur->next[s[j++] - 'a'];
                if(cur->isLeaf){
                    dp[j].push_back(i);
                    //表示以j结尾的前缀单词的开始下标在i
                }
            }
        }
        //得到dp数组之后，从最后一个字母开始向前DFS，最终可以搜索到所有的单词。
        vector<string> ans;
        vector<string> tWord;
        DFS(s,ans,tWord,dp,l);
        return ans;
    }
};