[백준] 수리공 항승(1449) - python

✏️ 문제

항승이는 품질이 심각하게 나쁜 수도 파이프 회사의 수리공이다. 항승이는 세준 지하철 공사에서 물이 샌다는 소식을 듣고 수리를 하러 갔다.
파이프에서 물이 새는 곳은 신기하게도 가장 왼쪽에서 정수만큼 떨어진 거리만 물이 샌다.
항승이는 길이가 L인 테이프를 무한개 가지고 있다.
항승이는 테이프를 이용해서 물을 막으려고 한다. 항승이는 항상 물을 막을 때, 적어도 그 위치의 좌우 0.5만큼 간격을 줘야 물이 다시는 안 샌다고 생각한다.
물이 새는 곳의 위치와, 항승이가 가지고 있는 테이프의 길이 L이 주어졌을 때, 항승이가 필요한 테이프의 최소 개수를 구하는 프로그램을 작성하시오. 테이프를 자를 수 없고, 테이프를 겹쳐서 붙이는 것도 가능하다.

🎈 입력

첫째 줄에 물이 새는 곳의 개수 N과 테이프의 길이 L이 주어진다. 둘째 줄에는 물이 새는 곳의 위치가 주어진다. N과 L은 1,000보다 작거나 같은 자연수이고, 물이 새는 곳의 위치는 1,000보다 작거나 같은 자연수이다.

🎈 출력

첫째 줄에 항승이가 필요한 테이프의 개수를 출력한다.

🎈 입출력 예

<입력>
4 2
1 2 100 101

<출력>
2

<입력>
4 3
1 2 3 4

<출력>
2

<입력>
3 1
3 2 1

<출력>
3

👩‍💻 내 코드

처음 문제를 보았을 땐 엄청 복잡할 것 같다고 생각했는데 예상외로 쉬웠다. 나는 아래를 중심으로 코드를 구현했다.

1. 처음 물새는 곳에서 테이프 길이 범위 안에 존재하는 곳
2. 테이프 길이 (길이가 1인 경우)

자세한 설명은 코드 주석으로 적어놓았다.

N, L = map(int, input().split())

water = list(map(int, input().split())) # 전체 물 새는 곳
water.sort(reverse=True) # 뒤에서부터 꺼내기 때문에 reverse

result = 0 # 결과 값
use_tape = L-1 # 사용한 테이프
tmp = water.pop() # 처음 물 새는 곳

while water:
    tmp1 = water.pop() # 그 다음 물 새는 곳
    if tmp1 < (tmp+L): # 처음 물 새는 곳 범위 안에 존재할 경우
        use_tape -= (tmp-tmp1) # 테이프 사용
        if use_tape == 0: # 테이프를 다 사용했을 경우
            result += 1 
            use_tape = L-1 # 결과 값에 1 더해주고 원래 길이로 할당
    else: # 처음 물 새는 곳 범위 안에 존재하지 않는 경우
        tmp = tmp1  # 다음 구간 처음 물 새는 곳으로 할당
        if L == 1 or use_tape != 0: # 가지고 있는 테이프 길이가 1이거나 사용한 테이프 길이가 0이 아닐 때
            use_tape = L-1
            result += 1

if L == 1 or use_tape != 0:
    result += 1

print(result)