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 {
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; } }
|