Softeer - 스마트 물류 (Python)

Lv3, ⭐⭐⭐

문제 풀이

• 어렵지 않았다. 3~5분정도 걸린듯
• 방문하고 있는 인덱스가 로봇일 때, 해당 로봇의 -k ~ +k위치의 부품을 찾고, 해당 부품을 방문처리 해주면 끝

from sys import stdin
n, k = map(int, stdin.readline().split())
line = list(map(str, stdin.readline().rstrip()))
# P 로봇, H 부품

res = 0
visited = [0] * n
for i in range(n):
    if line[i] == 'P':
        for val in range(i - k, i + k + 1):
            if n > val >= 0:
                if line[val] == 'H' and not visited[val]:
                    res += 1
                    visited[val] = 1
                    break
print(res)