【哈希-中等】438. 找到字符串中所有字母异位词

【题目】
给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。
字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。
说明:
字母异位词指字母相同,但排列不同的字符串。
不考虑答案输出的顺序。
【示例 1】
输入:
s: “cbaebabacd” p: “abc”
输出:
[0, 6]
解释:
起始索引等于 0 的子串是 “cba”, 它是 “abc” 的字母异位词。
起始索引等于 6 的子串是 “bac”, 它是 “abc” 的字母异位词。
【示例 2】
输入:
s: “abab” p: “ab”
输出:
[0, 1, 2]
解释:
起始索引等于 0 的子串是 “ab”, 它是 “ab” 的字母异位词。
起始索引等于 1 的子串是 “ba”, 它是 “ab” 的字母异位词。
起始索引等于 2 的子串是 “ab”, 它是 “ab” 的字母异位词。
【代码】
【Python】
在这里插入图片描述
这个评判系统也是蛮神奇的,昨天还可以通过,今天就是超时了。。。

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        s+=","
        p="".join(sorted(list(p)))
        pos=0
        ans=[]
        for index,c in enumerate(s):
            if index-pos==len(p):
                if "".join(sorted(list(s[pos:index])))==p:
                    ans.append(pos)
                pos+=1
        return ans

滑动窗口+数组
在这里插入图片描述

  1. 因为字符串中的字符全是小写字母,可以用长度为26的数组记录字母出现的次数
    设n = len(s), m = len§。
  2. 记录p字符串的字母频次p_cnt,和s字符串前m个字母频次s_cnt
  3. 若p_cnt和s_cnt相等,则找到第一个异位词索引 0
  4. 继续遍历s字符串索引为[m, n)的字母,在s_cnt中每次增加一个新字母,去除一个旧字母
  5. 判断p_cnt和s_cnt是否相等,相等则在返回值res中新增异位词索引 i - m + 1
class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        len_s,len_p,ans,p_cnt,s_cnt=len(s),len(p),[],[0]*26,[0]*26
        if len_s<len_p:
            return ans
        for i in range(len_p):
            p_cnt[ord(p[i])-ord('a')]+=1
            s_cnt[ord(s[i])-ord('a')]+=1
        if p_cnt==s_cnt:
            ans.append(0)
        for i in range(len_p,len_s):
            s_cnt[ord(s[i])-ord('a')]+=1
            s_cnt[ord(s[i-len_p])-ord('a')]-=1
            if s_cnt==p_cnt:
                ans.append(i-len_p+1)
        return ans

【滑动窗口+双指针】
在这里插入图片描述

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        n, m, res = len(s), len(p), []
        if n < m: return res
        p_cnt = [0] * 26
        s_cnt = [0] * 26

        for i in range(m):
            p_cnt[ord(p[i]) - ord('a')] += 1
        
        left = 0
        for right in range(n):
            cur_right = ord(s[right]) - ord('a')
            s_cnt[cur_right] += 1
            while s_cnt[cur_right] > p_cnt[cur_right]:
                cur_left = ord(s[left]) - ord('a')
                s_cnt[cur_left] -= 1
                left += 1
            if right - left + 1 == m:
                res.append(left)
        return res

猜你喜欢

转载自blog.csdn.net/kz_java/article/details/115016993
今日推荐