题目:
给定一个字符串 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"]输出:[]来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/substring-with-concatenation-of-all-words著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
答案:
时间:
双指针不愧是我最菜的算法,看不出来不说,知道了双指针,还经常写错。
class Solution:def findSubstring(self, s: str, words: List[str]) -> List[int]:if not s:return []if not words:return []n,k,m=len(words),len(words[0]),len(s)ans=[]for p in range(k):record= collections.Counter(words)i,j=p,p+kwhile j<=m:now_word=s[j-k:j]# print(i,j,record,now_word)if now_word not in record:i=jj=i+krecord= collections.Counter(words)else:if record[now_word]>0 and j-i==n*k:ans.append(i)i+=krecord[s[i-k:i]]+=1while record[now_word]<=0:i+=krecord[s[i-k:i]]+=1j+=krecord[now_word]-=1return ans
要点:
1. 对0-K分别做一次双指针
2. 如果这个单词没找到
3. 如果这个单词还有,并且长度差满足了nk
答案入i,ij后移,并且不会经过while循环,当前单词减1,因为i右移,所以前面单词+1
4. 如果这个单词没有了,不断右移i,直到找回来(一定能找到)
i 右移有限次,前缀单词有限次+1,找回单词后,j右移,当前单词-1。
