连接词

// Java Stream Solution
// 思路：与Java解决方案类似，我们使用一个Set来存储所有的单词。我们遍历每一个单词，
// 使用Stream和递归方法来检查它是否可以由其他单词组成。为了防止自我匹配，我们在检查时
// 临时从Set中移除当前单词。
 
import java.util.*;
import java.util.stream.Collectors;
 
public class StreamSolution {
    private Set<String> wordSet;
 
    public List<String> findAllConcatenatedWordsInADict(String[] words) {
        wordSet = Arrays.stream(words).collect(Collectors.toSet());
        return Arrays.stream(words)
                .filter(this::canForm)
                .collect(Collectors.toList());
    }
 
    private boolean canForm(String word) {
        if (wordSet.contains(word)) return true;
 
        return IntStream.range(1, word.length())
                .mapToObj(i -> new String[] {word.substring(0, i), word.substring(i)})
                .anyMatch(parts -> wordSet.contains(parts[0]) && canForm(parts[1]));
    }
}