[Python] 기지국 설치

코딩테스트 연습 - 기지국 설치

풀이

1. 1의 위치에서 부터 기지국이 전파를 전달하는 거리에 위치해 있는지 판단.

2. 범위 안에 있다면 locate를 현재 기지국의 오른쪽 방향으로 거리 밖으로 이동 시키고 idx도 다음 기지국을 가리키도록 1 증가

 

def solution(n, stations, w):
    answer = 0
    idx = 0
    locate = 1
    while locate <= n:
        if idx < len(stations) and locate >= stations[idx]-w:
            locate += stations[idx] + w + 1
            idx += 1
        else:
            locate = 2*w + 1
            answer += 1

    return answer