LeeCode第30题串联所有单词的子串

给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。

 s 中的 串联子串 是指一个包含  words 中所有字符串以任意顺序排列连接起来的子串。

例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab" 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。
返回所有串联字串在 s 中的开始索引。你可以以 任意顺序 返回答案。

 题目测试链接

示例 1：

输入：s = "barfoothefoobarman", words = ["foo","bar"]
输出：[0,9]
解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
输出顺序无关紧要。返回 [9,0] 也是可以的。

示例 2：

输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
输出：[]
解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
所以我们返回一个空数组。

示例 3：

输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出：[6,9,12]
解释：因为 words.length == 3 并且 words[i].length == 3，所以串联子串的长度必须为 9。
子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。

提示：

1 <= s.length <= 104
1 <= words.length <= 5000
1 <= words[i].length <= 30
words[i] 和 s 由小写英文字母组成

解题思路

这个问题的word长度小于5000，可用动态规划来解决

先来看动态规划的函数process

List<Integer> result - 存结果的，存放每个字符串开始的位置

String s - 原始需要拼接的字符串

int start - 当前验证到字符串的哪个位置

ArrayList<String> visit - 有多少个可以用来拼接的小串

int initial - 当前字符串开始拼接的位置

int len - 每一个小串的长度，这里每个小串都是等长的

java实现

class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        // visit 所有的小串，就是words中的所有小串都放进去
        ArrayList<String> visit = new ArrayList<>();
        // words 中小串的去重结果
        HashSet<String> sets = new HashSet<>();
        for(int i=0; i<words.length; i++){
            visit.add(words[i]);
            sets.add(words[i]);
        }
        int len = words[0].length();
        // 优化条件：words中的小串连起来比s长，就直接返回空的列表
        if(len * words.length > s.length()){
            return new ArrayList<>();
        }
        // 优化条件，words中只有一个小串
        // 就把words中的小串不考虑顺序全部链接成一个串，放在visit中
        if(sets.size()==1){
            String tmp = "";
            for(int i=0; i<words.length; i++){
                tmp += words[i];
            }
            visit.clear();
            visit.add(tmp);
            len = tmp.length();
        }
        List<Integer> result = new ArrayList<>();
        // 从0位置开始做尝试
        for(int i=0; i<s.length()-len+1; i++){
            process(result, s, i, visit, i, len);
        }
        return result;
    }
    // result - 存结果的，存放每个字符串开始的位置
    // s - 原始需要拼接的字符串
    // start - 当前验证到字符串的哪个位置
    // visit - 有多少个可以用来拼接的小串
    // initial - 当前字符串开始拼接的位置
    // len - 每一个小串的长度，这里每个小串都是等长的
    public void process(List<Integer> result, String s, 
            int start, ArrayList<String> visit, int initial, int len){
        // 所有的小串都用完了，就把initial加到最终的结果里面去
        if(visit.isEmpty()){
            result.add(initial);
        }
        // 已经到达字符串的最终位置，长度已经不够了，终止尝试
        if(start==s.length()){
            return;
        }
        // 尝试字符串的start~start+len 位置是否在visit当中
        if(start+len<=s.length()&&visit.contains(s.substring(start, start+len))){
            visit.remove((String) s.substring(start, start+len));
            // 在visit中就尝试下一个start+len位置
            process(result, s, start+len, visit, initial, len);
            visit.add(s.substring(start, start+len));
        }
    }
}

python

class Solution(object):
    def findSubstring(self, s, words):
        """
        :type s: str
        :type words: List[str]
        :rtype: List[int]
        """
        # visit 所有的小串，就是words中的所有小串都放进去
        visit = []
        # words 中小串的去重结果
        sets = set()
        for i in range(len(words)):
            visit.append(words[i])
            sets.add(words[i])
        len_ = len(words[0])
        # 优化条件：words中的小串连起来比s长，就直接返回空的列表
        if len_ * len(words) > len(s):
            return []
        # 优化条件，words中只有一个小串
        # 就把words中的小串不考虑顺序全部链接成一个串，放在visit中
        if len(sets) == 1:
            tmp = ""
            for i in range(len(words)):
                tmp += words[i]
            visit = []
            visit.append(tmp)
            len_ = len(tmp)
        result = []
        for i in range(len(s) - len_ + 1):
            self.process(result, s, i, visit, i, len_)
        return result
    # result - 存结果的，存放每个字符串开始的位置
    # s - 原始需要拼接的字符串
    # start - 当前验证到字符串的哪个位置
    # visit - 有多少个可以用来拼接的小串
    # initial - 当前字符串开始拼接的位置
    # len_ - 每一个小串的长度，这里每个小串都是等长的
    def process(self, result, s, start, visit, initial, len_):
        # 所有的小串都用完了，就把initial加到最终的结果里面去
        if len(visit) == 0:
            result.append(initial)
        # 已经到达字符串的最终位置，长度已经不够了，终止尝试
        if start == len(s):
            return
        # 尝试字符串的start~start+len 位置是否在visit当中
        if start + len_ <= len(s) and s[start:start+len_] in visit:
            visit.remove(s[start:start+len_])
            # 在visit中就尝试下一个start+len位置
            self.process(result, s, start+len_, visit, initial, len_)
            visit.append(s[start:start+len_])