首页 > 代码库 > 【leetcode】Word Break II
【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"。
代码如下:
1 public class Solution { 2 public List<String> wordBreak(String s, Set<String> dict) { 3 Map<String, ArrayList<String>> map = new HashMap<String, ArrayList<String>>(); 4 return findList(s, dict, map); 5 } 6 private List<String> findList(String s,Set<String> dict,Map<String, ArrayList<String>> map){ 7 if(map.containsKey(s)) return map.get(s); 8 ArrayList<String> answerList = new ArrayList<String>(); 9 int length = s.length();10 if(length <= 0)11 return answerList;12 13 for(int i = 1;i <= length;i++){14 String prefix = s.substring(0,i);15 if(dict.contains(prefix)){16 if(i == length)17 answerList.add(prefix);18 else{ 19 List<String> temp = findList(s.substring(i), dict, map);20 for(String tmp:temp){21 tmp = prefix + " " + tmp;22 answerList.add(tmp);23 }24 }25 }26 }27 map.put(s, answerList);28 return answerList;29 }30 }