Word Break II

阅读更多

转自：http://www.acmerblog.com/word-break-ii-6128.html?utm_source=tuicool

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"].

请参考LeetCode-Word Break 查看该题的第一个版本：判断是否可分词成功。但是这一题，不仅要判断是否可以分词，还要找到所有的分词结果。

第一题是用dp[i]表示 字符串s[0....i-1]是否可以分词成功。

为了记录是怎么分词的，我们可以用dp[i]表示 字符串s[0.....i-1]的最后一个分词word，则s[0.....i-1]的倒数第二个分词即为dp[o....i-word.length()-1]。但这种方法只能记录一种分词方法，因此有必要把 dp[i] 定义 List，这样就可以表示多种分词方法，最后使用DFS搜索，找到所有的分词方法。

例如对于上面的例子:  s.length()=10, 则 dp[10] = {“cat”}, dp[7]={“and”,”sand”}, dp[4]={“cats”}, dp[3]={“cat”}, dp[0]={}, 其它的都为null。

dfs搜索的时，可以从后向前，也可以从前向后，只要保证最终结果的顺序即可。下面是用的从后向前搜索。

01 public class Solution {
02     public static List wordBreak(String s, Set dict) {
03         List dp[] = new ArrayList[s.length()+1];
04         dp[0] = new ArrayList();
05         for(int i=0; i
06             //i是开始位置
07             if( dp[i] == null ) continue; //前面的部分必须是可以匹配的
08             for(String word:dict){
09                 int len = word.length();
10                 int end = i+len;
11                 if(end > s.length()) continue;
12                 if(s.substring(i,end).equals(word)){
13                     if(dp[end] == null){
14                         dp[end] = new ArrayList();
15                     }
16                     dp[end].add(word);//记录上一个位置
17                 }
18             }
19         }
20
21         List ans = new LinkedList();
22         if(dp[s.length()] == null) return ans;
23         ArrayList tmp = new ArrayList();
24         dfsStringList(dp,s.length(),ans, tmp);
25         return ans;
26     }
27
28     public static void dfsStringList(List dp[],int end,List res, ArrayList tmp){
29         if(end <= 0){
30             String ans = tmp.get(tmp.size()-1);
31             for(int i=tmp.size()-2; i>=0; i--)
32                 ans += (" " + tmp.get(i) );
33             res.add(ans);
34             return;
35         }
36         for(String str:dp[end]){
37             tmp.add(str);
38             dfsStringList(dp,end-str.length(), res, tmp);
39             tmp.remove(tmp.size()-1);
40         }
41     }
42 }

当然也有其他的解法，这个只是参考，效率其实还可以。