BOJ 19941 - 햄버거 분배

문제

백준 온라인 저지 - 19941번

풀이 과정

햄버거와 사람의 위치가 주어질 때 가장 많은 햄버거를 먹는 방법을 찾는 그리디 문제입니다.

일정 범위내의 햄버거를 먹을 수 있기 때문에 왼쪽부터 오른쪽으로 하나씩 먹을 수 있는 햄버거를 찾으면 됩니다.

코드

import sys

N, K = list(map(int, sys.stdin.readline().split()))
table = sys.stdin.readline().strip()


def solution():
    eat = [0] * N
    answer = 0

    for i in range(N):
        if table[i] == 'P':
            for offset in range(-K, K + 1):
                ni = i + offset
                if 0 <= ni < N and table[ni] == 'H' and not eat[ni]:
                    eat[ni] = 1
                    answer += 1
                    break

    return answer


print(solution())