개발자 톡

연습문제 톡 스마트 물류

스마트 물류(414번)

등록일
2023-05-27 15:48:39
조회수
494
작성자
sun17789

def main():
    N, K = map(int, input().split())
    rail = list(map(str, input()))

    answer = 0
    visited = [0] * (N+1)

    for i in range(N):

        for j in range(K, 0, -1):
            if i+j >= N:
                continue

            if visited[i] == 0 and visited[i+j] == 0:

                if rail[i] == 'P' and rail[i+j] == 'H':
                    visited[i] = 1
                    visited[i+j] = 1
                    answer += 1

                if rail[i] == 'H' and rail[i+j] == 'P':
                    visited[i] = 1
                    visited[i+j] = 1
                    answer += 1
    return answer

print(main())


오답처리가 되었는데 혹시 어느 부분이 테스트 케이스에 걸렸을지 궁굼하며 테스트 케이스 문제에 주어진 거 말고 다른 테스트 케이스 알려주실 수 있나요 ?

#스마트_물류
#python

이 카테고리의 톡 더보기