-
Notifications
You must be signed in to change notification settings - Fork 0
/
5H. Substring.py
105 lines (80 loc) · 2.41 KB
/
5H. Substring.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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
"""
В этой задаче Вам требуется найти максимальную по длине подстроку данной строки,
такую что каждый символ встречается в ней не более k раз.
Формат ввода
В первой строке даны два целых числа n и k (1 ≤ n ≤ 100000, 1 ≤ k ≤ n ) , где n
– количество символов в строке. Во второй строке n символов – данная строка, состоящая
только из строчных латинских букв.
Формат вывода
В выходной файл выведите два числа – длину искомой подстроки и номер её первого
символа. Если решений несколько, выведите любое.
Пример 1
Ввод
Вывод
3 1
abb
2 1
Пример 2
Ввод
Вывод
5 2
ababa
4 1
"""
from collections import Counter
class MultiSet(Counter):
def add(self, key):
if key not in self:
self[key] = 0
self[key] += 1
def pop(self, key):
if key in self:
if self[key] == 1:
super().pop(key)
else:
self[key] -= 1
def fun(s, k):
"""
>>> fun("abb", 1)
(2, 1)
>>> fun("ababa", 2)
(4, 1)
>>> fun("aaaaaaaaaa", 20)
(10, 1)
>>> fun("aassddaassddr", 2)
(7, 7)
"""
def update_max(left, right, max_len, ans):
cur_len = right - left
if cur_len > max_len:
max_len = cur_len
ans = (left, right - 1)
return ans, max_len
else:
return ans, max_len
d = MultiSet()
max_len = 0
ans = None
right = 0
for left in range(len(s)):
d.pop(s[left - 1])
while right < len(s):
if d[s[right]] == k:
ans, max_len = update_max(left, right, max_len, ans)
break
else:
d.add(s[right])
right += 1
else:
ans, max_len = update_max(left, right, max_len, ans)
break
left, right = ans
return right - left + 1, left + 1
n, k = map(int, input().split())
s = input()
print(*fun(s, k))
# print(f">>> fun(\"{s}\", {k})")
# print(f"{fun(s, k)}")
if __name__ == "__main__":
import doctest
doctest.testmod()