给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict,在字符串中增加空格来构建一个句子,使得句子中所有的单词都在词典中。返回所有这些可能的句子。
说明:
- 分隔时可以重复使用字典中的单词。
- 你可以假设字典中没有重复的单词。
示例 1:
输入: s = "catsanddog
" wordDict =["cat", "cats", "and", "sand", "dog"]
输出:[ "cats and dog", "cat sand dog" ]
思路:
本题要打印出所有单词组合,因此要进行dfs,但是仅仅使用dfs会超时,因此要参考单词拆分1的思路,构建dn[i]数组表示从i到字符串末尾能被拆分成字典中的词,在dfs同时判断即将要进行dfs的剩余字符串是否能被拆分,不能则跳过
class Solution {
public List<String> wordBreak(String s, List<String> wordDict) {
Map<String,String> map = new HashMap<>();
for(String str:wordDict){
map.put(str," ");
}
boolean[] dn = new boolean[s.length()+1];
dn[s.length()]=true;
for(int i=s.length()-1;i>=0;i--){
for(int j=i+1;j<=s.length();j++){
String str = s.substring(i,j);
boolean con = map.containsKey(str);
dn[i]=dn[j]&con;
if(dn[i]){break;}
}
}
List<String> list = new ArrayList<>();
List<String> res = new ArrayList<>();
wordCut(s,map,0,list,res,dn);
return res;
}
public void wordCut(String s,Map<String,String> dict,int l,List<String> list,List<String> res,boolean[] dn){
if(l==s.length()){
StringBuilder sb = new StringBuilder();
for(int i=0;i<list.size();i++){
sb.append(list.get(i));
if(i!=list.size()-1){
sb.append(" ");
}
}
res.add(sb.toString());
return ;
}
for(int i=l+1;i<=s.length();i++){
String tmp = s.substring(l,i);
//System.out.println(tmp);
if(dict.containsKey(tmp)&&dn[i]){
list.add(tmp);
wordCut(s,dict,i,list,res,dn);
list.remove(list.size()-1);
}
}
return ;
}
}