-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy patha-block.py
63 lines (52 loc) · 2.31 KB
/
a-block.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
# Copyright Sergey Lagov 2022 [email protected]
class Blocker:
def __init__(self, n: int, p: int, b: int, b_max: int, cur: int):
self.n = n
self.p = p
self.b_min = b
self.b_max = b_max
self.cur = cur
self.tries = []
def add_try(self, failure: int):
if 2 * self.b_max >= self.cur - failure:
self.tries.append(failure)
def analyze(self) -> (bool, int):
self.tries.sort()
b_cur, blocked_before, block_start, try_num = self.b_min, False, 0, 0
while try_num <= len(self.tries) - self.n:
if self.tries[try_num + self.n - 1] - self.tries[try_num] >= self.p:
try_num += 1
continue
if blocked_before:
b_cur *= 2
b_cur = self.b_max if b_cur > self.b_max else b_cur
try_num += self.n
block_start, blocked_before = self.tries[try_num - 1], True
if block_start + b_cur < self.cur or not blocked_before:
return False, None
return True, block_start + b_cur
def main():
"""
В условии задачи ничего не сказано о инкапсуляции — за её соблюдением я не следил
Сперва мы записываем наши n попыток в массив — O(n)
Затем сортируем массив размера n по возрастанию — O(n*logn)
Далее совершаем проход по массиву для выяснения статуса блокировки — O(n)
=> Время работы алгоритма — O(n + n*logn + n) = O(n*logn)
Согласно документации, sort в Python — Timsort. Сложность Timsort по памяти — O(n)
Хранение массива с попытками обойдётся в O(n)
=> Сложность алгоритма по памяти — O(n)
"""
try:
n, p, b, b_max, cur_unix = map(int, input().split())
except Exception:
return
blocker = Blocker(n, p, b, b_max, cur_unix)
while True:
try:
blocker.add_try(int(input()))
except EOFError:
break
status, time = blocker.analyze()
print(time if status else 'ok')
if __name__ == '__main__':
main()