Java实现 LeetCode 140 单词拆分 II（二）

140. 单词拆分 II

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，在字符串中增加空格来构建一个句子，使得句子中所有的单词都在词典中。返回所有这些可能的句子。

说明：

分隔时可以重复使用字典中的单词。

你可以假设字典中没有重复的单词。

示例 1：

输入:

s = “catsanddog”

wordDict = [“cat”, “cats”, “and”, “sand”, “dog”]

输出:

[

“cats and dog”,

“cat sand dog”

]

示例 2：

输入:

s = “pineapplepenapple”

wordDict = [“apple”, “pen”, “applepen”, “pine”, “pineapple”]

输出:

[

“pine apple pen apple”,

“pineapple pen apple”,

“pine applepen apple”

]

解释: 注意你可以重复使用字典中的单词。

示例 3：

输入:

s = “catsandog”

wordDict = [“cats”, “dog”, “sand”, “and”, “cat”]

输出:

[]

class Solution {
      private Map<String, List<String>> cache = new HashMap<>();

    public List<String> wordBreak(String s, List<String> wordDict) {
        return dfs(s, wordDict,0);
    }

    private List<String> dfs(String s, List<String> wordDict, int offset){

        if (offset == s.length()){
            List<String> res = new ArrayList<>();
            res.add("");
            return res;
        }

        if (cache.containsKey(s.substring(offset))){
           return cache.get(s.substring(offset));
        }

        List<String> res = new ArrayList<>();
        for (String word : wordDict){
            if (word.equals(s.substring(offset, Math.min(s.length(),offset + word.length())))){
                List<String> next = dfs(s, wordDict, offset + word.length());
                for (String str: next){
                    res.add((word + " "+ str).trim());
                }
            }
        }

        cache.put(s.substring(offset),res);
        return res;
    }
}