Leetcode 472.连接词

连接词

给定一个不含重复单词的列表，编写一个程序，返回给定单词列表中所有的连接词。

连接词的定义为：一个字符串完全是由至少两个给定数组中的单词组成的。

示例:

输入: ["cat","cats","catsdogcats","dog","dogcatsdog","hippopotamuses","rat","ratcatdogcat"]

输出: ["catsdogcats","dogcatsdog","ratcatdogcat"]

解释: "catsdogcats"由"cats", "dog" 和 "cats"组成;

"dogcatsdog"由"dog", "cats"和"dog"组成;

"ratcatdogcat"由"rat", "cat", "dog"和"cat"组成。

说明:

1. 给定数组的元素总数不超过 10000。
2. 给定数组中元素的长度总和不超过 600000。
3. 所有输入字符串只包含小写字母。
4. 不需要考虑答案输出的顺序。

思路是：

对于words中的每个单词w，我们定义一个数组dp[n+1]，如果dp[i] == true，则表示w.substr(0, i)可以由words中的已有单词连接而成。那么状态转移方程就是：dp[i] = {dp[j] && w.substr(j + 1, i - j) is in words}，其中j < i。最终检查dp[n]是否为true，如果是则将其加入结果集中。为了加速对words中的单词的查找，我们用一个哈希表来保存各个单词。这样时间复杂度可以降低到O(n * m^2)，其中n是words中的单词的个数，m是每个单词的平均长度（或者最大长度？）。

import java.util.*;

class Solution {
    public List<String> findAllConcatenatedWordsInADict(String[] words) {
        Set<String> set=new HashSet<String>();
        for(int i=0;i<words.length;i++){
            set.add(words[i]);
        }
        List<String> res=new ArrayList<String>();
        for(String word:words){
            int n=word.length();
            boolean[] dp=new boolean[n+1];
            Arrays.fill(dp,false);
            dp[0]=true;
            for(int i=0;i<n;i++){
                if(!dp[i]) continue;
                for(int j=i+1;j<=n;j++){
                    if(j-i<n && set.contains(word.substring(i,j))){
                        dp[j]=true;
                    }
                }
                if(dp[n]){
                    res.add(word);
                    break;
                }
            }
        }
        return res;
    }
}