여니의 취준 준비/코딩테스트 (Python)

[n19941] 햄버거 분배 in python

여니's 2021. 10. 30. 16:37


최대한 왼쪽에 있는 값을 선택하게끔 해줘야 한다.

사실 idx 처리하는 부분에 있어서

고민을 좀 했는데

for문 안에 max, min 함수를 이용하면

손쉽게 처리가 가능하다는 것을 알게 되었다.

 

 

n, k = map(int, input().split())
array = list(input())
answer = 0

for i in range(n):
    if array[i] == "P":  # 사람
        for j in range(max(i - k, 0), min(n, i + k + 1)):
            if array[j] == "H":
                answer += 1
                array[j] = 0
                break

print(answer)