19941. 햄버거 분배

19941 문제 보기

접근 방식

그냥 구현!
범위 체크만 잘 해주면 됨!

최종 코드

import sys
input = sys.stdin.readline

N, K = map(int, input().rstrip().split())
arr = list(input().rstrip())
ans = 0

for i in range(N):
  if arr[i] == 'P':
    for j in range(max(i - K, 0), min(i + K + 1, N)):
      if arr[j] == 'H':
        arr[j] = 0
        ans += 1
        break

print(ans)