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
}
}