首页 > 代码库 > Word Break II 求把字符串拆分为字典里的单词的全部方案 @LeetCode

Word Break II 求把字符串拆分为字典里的单词的全部方案 @LeetCode

这道题相似  Word Break 推断能否把字符串拆分为字典里的单词 @LeetCode 只不过要求计算的并不不过能否拆分,而是要求出全部的拆分方案。

因此用递归。

可是直接递归做会超时,原因是LeetCode里有几个非常长可是无法拆分的情况。所以就先跑一遍Word Break,先推断能否拆分。然后再进行拆分。


递归思路就是,逐一尝试字典里的每个单词,看看哪一个单词和S的开头部分匹配,假设匹配则递归处理S的除了开头部分,直到S为空。说明能够匹配。


Given a string s and a dictionary of words dict, add spaces in s to construct a sentence where each word is a valid dictionary word.

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"].


public class Solution {
    public List<String> wordBreak(String s, Set<String> dict) {
        List<String> list = new ArrayList<String>();
        List<String> ret = new ArrayList<String>();
        rec(s, dict, list, ret);
        return ret;
    }
    
    public void rec(String s, Set<String> dict, List<String> list, List<String> ret) {
        if(!isBreak(s, dict)){  // test before run to avoid TLE
            return;
        }
        if(s.length() == 0) {
            String concat = "";
            for(int i=0; i<list.size(); i++) {
                concat += list.get(i);
                if(i != list.size()-1) {
                    concat += " ";
                }
            }
            ret.add(concat);
            return;
        }
        
        for(String cur : dict) {
        	if(cur.length() > s.length()) {     // avoid out of boundary
        		continue;
        	}
            String substr = s.substring(0, cur.length());
            if(substr.equals(cur)) {
                list.add(substr);
                rec(s.substring(cur.length()), dict, list, ret);
                list.remove(list.size()-1);
            }
        }
    }
    
    public boolean isBreak(String s, Set<String> dict) {
        boolean[] canBreak = new boolean[s.length()+1];
        canBreak[0] = true;
        
        for(int i=1; i<=s.length(); i++) {
            boolean flag = false;
            for(int j=0; j<i; j++) {
                if(canBreak[j] && dict.contains(s.substring(j,i))) {
                    flag = true;
                    break;
                }
            }
            canBreak[i] = flag;
        }
        return canBreak[s.length()];
    }
}







Word Break II 求把字符串拆分为字典里的单词的全部方案 @LeetCode