Word Break II——LeetCode

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

题目大意就是：给一个string，一个词典，把这个string根据词典构建出所有可能的组合。

我的解法就是预处理+DFS+剪枝。

1、首先用一个breakFlag数组，记录string中的各个位置为结尾是否是合法的分词末尾，以上面的样例来说，c,a都是不合法的分词结尾，t,s都是合法的分词结尾。

2、然后用DFS来搜全部可能的组合，如果最后正好分完这个string，说明是合法的分词方法，组合成句子然后添加到结果List中，有个小优化就是可以先把字典里的单词的长度放入一个array，分词的时候只需要遍历这个array就可以，比如上面的字典里单词长度只有{3,4}组成一个array，只需要遍历cat cats就可以继续往后遍历了。

Talk is cheap>>

package leetcode;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class WordBreakII {

    public static void main(String[] args) {
        Set<String> set = new HashSet<String>();
        set.add("cat");
        set.add("cats");
        set.add("and");
        set.add("sand");
        set.add("dog");
        new WordBreakII().wordBreak("catsanddog", set);
    }

    Set<Integer> lenArray = new HashSet<>();
    boolean[] breakFlag;

    public List<String> wordBreak(String s, Set<String> dict) {
        List<String> res = new ArrayList<>();
        for (String next : dict) {
            lenArray.add(next.length());
        }
        breakFlag = new boolean[s.length() + 1];
        breakFlag[0] = true;
        for (int i = 0; i < s.length(); i++) {
            if (breakFlag[i]) {
                for (int j = 0; i + j < s.length() + 1; j++) {
                    if (dict.contains(s.substring(i, i + j)))
                        breakFlag[i + j] = true;
                }
            }
        }
        if (breakFlag[s.length()])
            dfs(s, "", dict, res, s.length());
        return res;
    }

    public void dfs(String src, String tmp, Set<String> dict, List<String> res, int length) {
        if (length < 0) {
            return;
        }
        if (length == 0) {
            System.out.println(tmp.substring(0, tmp.length() - 1));
            res.add(tmp.substring(0, tmp.length() - 1));
            return;
        }
        for (int len : lenArray) {
            int left = src.length() - len;
            if (left < 0)
                break;
            String t = src.substring(left, src.length());
            if (breakFlag[length] && dict.contains(t)) {
                dfs(src.substring(0, left), t + " " + tmp, dict, res, length - len);
            }
        }
    }

}