-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsolution.py
47 lines (43 loc) · 1.39 KB
/
solution.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
class Solution(object):
def findAnagrams(self, s, p):
"""
:type s: str
:type p: str
:rtype: List[int]
"""
res = []
pattern_length = len(p)
pattern_map = {}
for c in p:
if pattern_map.get(c) == None:
pattern_map[c] = 0
pattern_map[c] += 1
copy_pattern_map = dict(pattern_map)
beg = 0
length = 0
i = 0
while i < len(s):
if length < pattern_length:
if copy_pattern_map.get(s[i]) == None:
copy_pattern_map = dict(pattern_map)
length = 0
i += 1
beg = i
elif copy_pattern_map[s[i]] > 0:
copy_pattern_map[s[i]] -= 1
length += 1
i += 1
else:
while s[beg] != s[i]:
copy_pattern_map[s[beg]] += 1
length -= 1
beg += 1
# in this place s[beg] == s[i], so we have to go forward 1 distance
beg += 1
i += 1
if length == pattern_length:
res.append(beg)
copy_pattern_map[s[beg]] += 1
length -= 1
beg += 1
return res