백준 알고리즘 19941. 햄버거 분배

2024. 3. 25. 20:21코딩 테스트/백준

https://www.acmicpc.net/problem/19941

 

19941번: 햄버거 분배

기다란 벤치 모양의 식탁에 사람들과 햄버거가 아래와 같이 단위 간격으로 놓여 있다. 사람들은 자신의 위치에서 거리가 $K$ 이하인 햄버거를 먹을 수 있다. 햄버거 사람 햄버거 사람 햄버거 사

www.acmicpc.net

 

전체 코드

김시은님 코드

N, K = map(int, input().split())
location = list(input())
count = 0

for i in range(N):
    if location[i] == 'P': # 현재 위치가 사람이라면 
        for j in range(max(i-K, 0), min(i+K+1, N)): # 범위 제한  
            # 해당 범위 내에서 햄버거를 찾았을 경우
            if location[j] == 'H':
                location[j] = 0 # 방문처리 
                count += 1 # 카운팅 
                break

print(count)