本文为Python算法题集之一的代码示例
题目438:找到字符串中所有字母异位词
说明:给定两个字符串 s
和 p
,找到 s
中所有 p
的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
示例 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" 的异位词。
提示:
-
1 <= s.length, p.length <= 3 * 104
-
s
和p
仅包含小写字母
问题分析
- 因p是固定的,所以检查是否为p的异位词可以直接使用数组的字符计数比较即可【p仅含小写字母,因此数组只有26个元素】
- 因p长度固定,因此单循环即可遍历字符串
- 优化思路
- 减少计算
- 加快比较
-
标准版【循环进行异位词比较】,性能良好,超越89%,标准版的性能就比较高,说明本题可以优化的空间不大
注意:
CheckFuncPerf
是我手搓的函数用时和内存占用模块,下载地址在这里:测量函数运行用时、内存占用的代码单元CheckFuncPerf.py以及使用方法
import CheckFuncPerf as cfp def findAnagrams(s: str, p: str) -> list[int]: list_p = [0] * 26 list_s = [0] * 26 list_result = [] for iIdx in range(len(p)): list_p[ord(p[iIdx])-ord('a')] += 1 for iIdx in range(len(s)): list_s[ord(s[iIdx])-ord('a')] += 1 if iIdx < len(p) - 1: continue if list_s == list_p: list_result.append(iIdx - len(p) + 1) list_s[ord(s[iIdx-len(p)+1]) - ord('a')] -= 1 return list_result s, p = 'cbaebabacd', 'abc' result = cfp.getTimeMemor