LeetCode30——串联所有单词的子串

题目描述

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。

示例 1：

输入：
s = “barfoothefoobarman”,
words = [“foo”,“bar”]
输出：[0,9]
解释：
从索引 0 和 9 开始的子串分别是 “barfoo” 和 “foobar” 。
输出的顺序不重要, [9,0] 也是有效答案。

示例 2：

输入：
s = “wordgoodgoodgoodbestword”,
words = [“word”,“good”,“best”,“word”]
输出: []

题目分析

首先用map记录每个word单词出现的次数。接着再用一个map记录words长度的字符串中找到了几个word，并且记录word出现的次数。
时间复杂度是O(M*N)，M是word的个数，N是字符串的长度。

代码

Java代码如下：

class Solution {
    public List<Integer> findSubstring(String s, String[] words) {
        List<Integer> result = new ArrayList<>();
        if(words.length==0)
        {
            return result;
        }
        Map<String,Integer> map = new HashMap<>();
        for(String word:words)
        {
            map.put(word,map.getOrDefault(word,0)+1);
        }
        int n=words.length;
        int len=words[0].length();
        Map<String,Integer> found = new HashMap<>();
        for(int i=0;i<=s.length()-n*len;i++)
        {
            found.clear();
            int j=0;
            for(j=0;j<n;j++)
            {
                String str =s.substring(i+j*len,i+j*len+len);
                if(map.containsKey(str))
                {
                    found.put(str,found.getOrDefault(str,0)+1);
                    if(found.get(str)>map.get(str))
                    {
                        break;
                    }
                } else  
                {
                    break;
                }
            }
            if(j==n)
            {
                result.add(i);
            }
        }
        return result;
    }
}