leetcode WordBreakII
z
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

public class WordBreakII {
/**
* Given a non-empty string s and a dictionary wordDict containing a list of non-empty words,
* add spaces in s to construct a sentence where each word is a valid dictionary word. Return all such possible sentences.
*
* Note:
*
* The same word in the dictionary may be reused multiple times in the segmentation.
* You may assume the dictionary does not contain duplicate words.
* Example 1:
*
* Input:
* s = "catsanddog"
* wordDict = ["cat", "cats", "and", "sand", "dog"]
* Output:
* [
* "cats and dog",
* "cat sand dog"
* ]
* Example 2:
*
* Input:
* s = "pineapplepenapple"
* wordDict = ["apple", "pen", "applepen", "pine", "pineapple"]
* Output:
* [
* "pine apple pen apple",
* "pineapple pen apple",
* "pine applepen apple"
* ]
* Explanation: Note that you are allowed to reuse a dictionary word.
* Example 3:
*
* Input:
* s = "catsandog"
* wordDict = ["cats", "dog", "sand", "and", "cat"]
* Output:
* []
*/
public List<String> wordBreak(String s, List<String> wordDict) {
HashMap<String, List<String>> map = new HashMap<>();
bfs( s, map, wordDict);
return map.get(s);
}

public List<String> bfs(String s, HashMap<String, List<String>> map, List<String> wordDick) {
if (map.containsKey(s)) {
return map.get(s);
}

List<String> t = new ArrayList<>();

if (s.isEmpty()) {
t.add("");
}

for (String word: wordDick) {
if (s.startsWith(word)) {
List<String> left = bfs( s.substring(word.length()), map, wordDick);
if (! left.isEmpty()) {
for (String l: left) {
t.add(word + (l.isEmpty()?"":" ") + l);
}
}
}
}
map.put(s, t);
return t;
}
}