Leetcode 140. Word Break II

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. You may assume the dictionary does not contain duplicate words.
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"].

题意：139的followup，求出所有的break方案。

思路：
第一想到的还是深度优先搜索，在139的基础上用一个String类型的path记录当前的break方案，再用一个List<String> res存储找到的所有path。

public List<String> wordBreak(String s, List<String> wordDict) {
    List<String> res = new ArrayList<>();
    if (s == null || s.length() == 0) {
        return res;
    }

    HashSet<String> set = new HashSet<>();
    for (String word : wordDict) {
        set.add(word);
    }

    dfs(s, 0, set, res, "");

    return res;
}

private void dfs(String s, int start, HashSet<String> set, List<String> res, String path) {
    if (start == s.length()) {
        res.add(path.trim());
        return;
    }

    for (int i = start + 1; i <= s.length(); i++) {
        String str = s.substring(start, i);
        if (!set.contains(str)) {
            continue;
        }
        dfs(s, i, set, res, path + str + " ");
    }
}

提交以后还是会超时，下面是一个记忆化搜索的解法，通过hashmap来记录后面子串已经搜索过的组合，来避免重复搜索。

public List<String> wordBreak(String s, Set<String> wordDict) {
    return DFS(s, wordDict, new HashMap<String, LinkedList<String>>());
}

// DFS function returns an array including all substrings derived from s.
List<String> DFS(String s, Set<String> wordDict, HashMap<String, LinkedList<String>>map) {
    if (map.containsKey(s))
        return map.get(s);

    LinkedList<String>res = new LinkedList<String>();
    if (s.length() == 0) {
        res.add("");
        return res;
    }
    for (String word : wordDict) {
        if (s.startsWith(word)) {
            List<String>sublist = DFS(s.substring(word.length()), wordDict, map);
            for (String sub : sublist)
                res.add(word + (sub.isEmpty() ? "" : " ") + sub);
        }
    }
    map.put(s, res);
    return res;
}