leetcode刷题笔记三十 串联所有单词的子串(待优化)

leetcode刷题笔记三十 串联所有单词的子串(待优化)

源地址:30. 串联所有单词的子串

问题描述:

给定一个字符串 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"]
输出:[]

代码补充:

//这里采用滑动窗口的方法
//由于words中单词长度相同,我们将单一单词的长度视为窗口的长度
//对于s字符串的s.length - wordLength长度的字符进行遍历
//判断以其开头的窗口内所含的子串是否在words中
//由于子串的串联,需要words中所有词汇出现一次,且每个词汇长度相同
//我们将words的长度视为窗口移动一次的距离,这样当前川口满足的情况下
//下次移动的结果即为下一个词汇
//值得注意的是,一开始我是使用集合的方式处理,但是这种情况下拥有相同词汇的words就无法通过了
//看了一些别人的解法,最终选择了Map对单词和次数进行映射,这样有效的解决了相同词汇的问题
//对初始的words做一个映射,对窗口获取的结果做一个映射,进行比较
//直接使用 " == " 比使用equals 慢了500ms左右
//思路和别人的scala解决方案基本一致,但是慢了3000ms,可能还有一些部分需要优化
object Solution {
    def findSubstring(s: String, words: Array[String]): List[Int] = {
        if (words.length == 0) return Nil
        val wordLength = words(0).length
        val totalLength = wordLength * words.length
        var temp = words
        var ans:List[Int] = List()
        val stringLength = s.length
        var firstMap:Map[String, Int] = Map.empty
        var countMap:Map[String, Int] = Map.empty
        words.foreach(elem => firstMap += (elem -> (firstMap.getOrElse(elem,0)+1) ) )
        //println(firstMap)

        for(i <- 0 to stringLength-wordLength){
            var temp = words
            if(words.contains(s.substring(i,i+wordLength))){
                words.foreach(elem => countMap += (elem -> 0) )
                //println("countMap: " + countMap)
                val start = i
                var left = i
                for(j <- 0 to words.length-1 if (left+wordLength <= s.length)){
                    if(temp.contains(s.substring(left,left+wordLength))){
                      val temp = s.substring(left,left+wordLength)
                      countMap = countMap + (temp ->(countMap.getOrElse(temp,0)+1))
                      left +=  wordLength   
                    }
                    //println("-------------------------------------")
                    //println("start: " + start)
                    //println("countMap: " + countMap)
                }
                if(countMap.equals(firstMap)) ans = ans :+ start
            }
           
        }
    return ans
    }
}
posted @ 2020-05-04 19:39  ganshuoos  阅读(144)  评论(0编辑  收藏  举报