Untitled

 avatar
unknown
plain_text
2 months ago
420 B
3
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