[Python] 백준 19941. 햄버거 분배

 

문제 링크

 

 

풀이

순회하면서 사람을 발견하면 양 옆으로 k거리만큼의 범위 내에서 햄버거를 찾고 count를 증가시킨 후 P와 H가 아닌 다른 문자로 바꿔준다.

 

코드

n, k = map(int, input().split())
a = list(input())
cnt = 0
for i in range(len(a)):
    if a[i] == 'P':
        for j in range(max(i-k,0),min(i+k+1,n)):
            if a[j] == 'H':
                cnt += 1
                a[j] = 'C'
                break
print(cnt)