Untitled
unknown
plain_text
9 months ago
420 B
4
Indexable
from typing import List
# Write any import statements here
def getMaxAdditionalDinersCount(N: int, K: int, M: int, S: List[int]) -> int:
# Write your code here
arr = [0] * N
res = 0
for ele in S:
pos = ele - 1
arr[pos - K : pos + K + 1] = [1] * (2*K + 1)
for i in range(len(arr)):
if arr[i] == 0:
res += 1
arr[i : i + K + 1] = [1] * (K + 1)
return res
Editor is loading...
Leave a Comment