LeetCode: Word Break II [140]

class Solution {
public:
    
    void recoverPath(vector<string>&result, vector<string>&words, string&s, map<int, vector<int> >preorders, int end){
        vector<int>preorder=preorders[end];
        for(int i=0; i<preorder.size(); i++){
            string word=s.substr(preorder[i]+1, end-preorder[i]);
            if(preorder[i]==-1){
                string cutstr=word;
                int size=words.size();
                for(int j=size-1; j>=0; j--){
                    cutstr+=" "+words[j];
                }
                result.push_back(cutstr);
            }
            else{
                words.push_back(word);
                recoverPath(result, words, s, preorders, preorder[i]);
                words.pop_back();
            }
        }
    }
    
    vector<string> wordBreak(string s, unordered_set<string> &dict) {
        vector<string> result;
        if(s.length()==0)return result;
        map<int, vector<int> >preorders;    //记录各个可分位置的前驱集合
        vector<int> pos(1,-1);         //以确定可分的位置
        
        for(int i=0; i<s.length(); i++){
            vector<int>preorder;
            for(int k=0; k<pos.size(); k++){
                if(dict.find(s.substr(pos[k]+1, i-pos[k]))!=dict.end()){
                    preorder.push_back(pos[k]);
                }
            }
            if(preorder.size()>0){
                preorders[i]=preorder;
                pos.push_back(i);
            }
        }
        
        //恢复所有可能的切分路径
        if(preorders.find(s.length()-1)==preorders.end())return result;
        vector<string>words;
        recoverPath(result, words, s, preorders, s.length()-1);
        return result;
    }
};