【leetcode】Word Break II

Given a string s and a dictionary of words dict, add spaces in s to construct a sentence where each word is a valid dictionary word.

Return all such possible sentences.

For example, given
s = "catsanddog",
dict = ["cat", "cats", "and", "sand", "dog"].

A solution is ["cats and dog", "cat sand dog"].

---

题解：开始想利用Word Break I的代码（详细请见http://www.cnblogs.com/sunshineatnoon/p/3849154.html），先把dp的表打出来，然后DFS搜索出字符串，后来发现这样会TLE。看了九章的题解，才有了如下思路：

递归的思想，利用一个map——Map<String, ArrayList<String>> map存放单词s和它对应的拆分字符串的列表。比如题目中的例子catsanddog在map中的对应的list就有两个元素"cats and dog"和"cat sand dog"。

因为每个字符串s如果有划分，那么一定从第一个字符开始，所以，递归每次从索引0开始，找到dict中包含的s的前缀，然后递归的探求s除去这个前缀后剩下的字符串可以被拆分成的方式，如果剩下的字符串可以被拆分，那么前缀加上剩下字符串拆分得到的结果就是最终s的拆分结果了。

还是以题目中的例子来说"catsanddog"

首先发现dict中含有前缀cat，这时递归的去找"sanddog"的拆分方式，发现可以拆分为”sand dog"，最终得到s的一种拆分方式"cat sand dog";

继续循环，发现dict中含有前缀"cats"，这时递归的去找"anddog"的拆分方式，发现可以拆分为"and dog"，最终得到s的第二种拆分方式"cats and dog"。

代码如下：

public class Solution {
    public List<String> wordBreak(String s, Set<String> dict) {
        Map<String, ArrayList<String>> map = new HashMap<String, ArrayList<String>>();
        return findList(s, dict, map);
    }
    private List<String> findList(String s,Set<String> dict,Map<String, ArrayList<String>> map){
        if(map.containsKey(s)) return map.get(s);
        ArrayList<String> answerList = new ArrayList<String>();
        int length = s.length();
        if(length <= 0)
            return answerList;
        
        for(int i = 1;i <= length;i++){
            String prefix = s.substring(0,i);
            if(dict.contains(prefix)){
                if(i == length)
                    answerList.add(prefix);
                else{                    
                    List<String> temp = findList(s.substring(i), dict, map);
                    for(String tmp:temp){
                        tmp = prefix + " " + tmp;
                        answerList.add(tmp);
                    }
                }
            }
        }
        map.put(s, answerList);
        return answerList;
    }
}