《代码随想录》学习笔记,原链接:代码随想录
class Solution:
def findAnagrams(self, s: str, p: str) -> List[int]:
if len(s) < len(p): # 特殊情况
return []
hash_s, hash_p = [0] * 26, [0] * 26 # 构建s和p的哈希数组
result = []
for i in range(len(p)): # 统计字符串p的字母和字符串s的前len(p)个字母
hash_p[ord(p[i]) - ord("a")] += 1
hash_s[ord(s[i]) - ord("a")] += 1
if hash_p == hash_s: # 判断0的位置是否为字母异位词
result.append(0)
for i in range(1, len(s) - len(p) + 1): # 从1位置开始向后遍历s
hash_s[ord(s[i - 1]) - ord("a")] -= 1 # 将i - 1位置的字母,从hash_s中剔除
hash_s[ord(s[i + len(p) - 1]) - ord("a")] += 1 # 将i + len(p) - 1位置的字母,加入到hash_s中
if hash_p == hash_s: # 判断此时是否为字母异位词
result.append(i)
return result
【注】
- 由于讨论的是字符串s中是否有字符串p,所以当len(s) < len(p)直接返回[]。
- for i in range(1, len(s) - len(p) + 1): ,遍历到每个位置,都需要移除i - 1位置的字母和增加i + len(p) - 1位置的字母,以保证hash_s中一直存储len(p)个字母的哈希表。