본문 바로가기
Algorithm/코딩테스트 (Python)

[백준][8980번] 택배

by 싱브이 2024. 1. 15.
728x90
반응형

문제

 

아래 그림과 같이 직선 도로상에 왼쪽부터 오른쪽으로 1번부터 차례대로 번호가 붙여진 마을들이 있다. 마을에 있는 물건을 배송하기 위한 트럭 한 대가 있고, 트럭이 있는 본부는 1번 마을 왼쪽에 있다. 이 트럭은 본부에서 출발하여 1번 마을부터 마지막 마을까지 오른쪽으로 가면서 마을에 있는 물건을 배송한다. 

각 마을은 배송할 물건들을 박스에 넣어 보내며, 본부에서는 박스를 보내는 마을번호, 박스를 받는 마을번호와 보낼 박스의 개수를 알고 있다. 박스들은 모두 크기가 같다. 트럭에 최대로 실을 수 있는 박스의 개수, 즉 트럭의 용량이 있다. 이 트럭 한대를 이용하여 다음의 조건을 모두 만족하면서 최대한 많은 박스들을 배송하려고 한다.

  • 조건 1: 박스를 트럭에 실으면, 이 박스는 받는 마을에서만 내린다.
  • 조건 2: 트럭은 지나온 마을로 되돌아가지 않는다.
  • 조건 3: 박스들 중 일부만 배송할 수도 있다.

마을의 개수, 트럭의 용량, 박스 정보(보내는 마을번호, 받는 마을번호, 박스 개수)가 주어질 때, 트럭 한 대로 배송할 수 있는 최대 박스 수를 구하는 프로그램을 작성하시오. 단, 받는 마을번호는 보내는 마을번호보다 항상 크다.

예를 들어, 트럭 용량이 40이고 보내는 박스들이 다음 표와 같다고 하자.

 

보내는 마을 받는 마을 박스 개수
1 2 10
1 3 20
1 4 30
2 3 10
2 4 20
3 4 20

이들 박스에 대하여 다음과 같이 배송하는 방법을 고려해 보자.

(1) 1번 마을에 도착하면

  • 다음과 같이 박스들을 트럭에 싣는다. (1번 마을에서 4번 마을로 보내는 박스는 30개 중 10개를 싣는다.)
보내는 마을 받는 마을 박스 개수
1 2 10
1 3 20
1 4 10

(2) 2번 마을에 도착하면

  • 트럭에 실려진 박스들 중 받는 마을번호가 2인 박스 10개를 내려 배송한다. (이때 트럭에 남아있는 박스는 30개가 된다.)
  • 그리고 다음과 같이 박스들을 싣는다. (이때 트럭에 실려 있는 박스는 40개가 된다.)
보내는 마을 받는 마을 박스 개수
2 3 10

(3) 3번 마을에 도착하면 

  • 트럭에 실려진 박스들 중 받는 마을번호가 3인 박스 30개를 내려 배송한다. (이때 트럭에 남아있는 박스는 10개가 된다.)
  • 그리고 다음과 같이 박스들을 싣는다. (이때 트럭에 실려 있는 박스는 30개가 된다.)
보내는 마을 받는 마을 박스 개수
3 4 20

(4) 4번 마을에 도착하면 

  • 받는 마을번호가 4인 박스 30개를 내려 배송한다

위와 같이 배송하면 배송한 전체 박스는 70개이다. 이는 배송할 수 있는 최대 박스 개수이다.

 

 

 

 

 

내 생각

 

문제가 길지만, 결국 문제가 원하는건 가장 많이 택배를 옮기고, 그 최대 개수를 구하는 것!

이 문제를 해결하기 위한 팁은 정렬 활용이다. 가까운 거리로 정렬해서 빠른 도착지부터 택배의 용량을 줄여가는 것이다.

 

일단 풀이 순서

1. 입력받기

2. 정렬 필! 수! (도착지 오름차순 정렬)

3. 출발지와 도착지 사이 반복문

4. 옮긴만큼 뺐던 박스 총 개수 반환

 

주석으로 적으면서 해결해서 주석에 설명이 되어 있다.

 

 

 

 

 

내 코드

 

# input 1 : 마을 수 N과 트럭의 용량 C
N, C = map(int, input().split())
# input 2 : 보내는 박스 정보의 개수 M
M = int(input())
# input 3 : 다음 M개의 박스 정보 -  박스를 보내는 마을번호(s), 박스를 받는 마을번호(r), 보내는 박스 개수(b)
boxes = [list(map(int, input().split())) for _ in range(M)]
# 도착지가 가까운 순으로 정렬 : 가까운 순으로 빠르게 운송하여 최대한 많이 운송!
boxes.sort(key=lambda x :  x[1])
# 지역마다 전달할 최대 박스 개수 (C개)
max_box = [C] * (N+1)
# output : 트럭 한 대로 배송할 수 있는 최대 박스 수
answer = 0

for s, r, b in boxes:
    _min = C
    # 출발지에서 도착지까지
    for i in range(s, r):
        # 옮길 수 있는 박스 개수(_min)와 지역 운송 최대 박스 개수(max_box) 중 작은 값
        _min = min(_min, max_box[i])
    # 옮길 수 있는 박스 개수(_min)와 실제 박스 개수(b) 중 작은 값
    _min = min(_min, b)
    for i in range(s, r):
        # 옮긴 만큼은 빼기 !
        max_box[i] -= _min
    # 뺐던 박스 개수
    answer += _min
print(answer)
728x90
반응형

'Algorithm > 코딩테스트 (Python)' 카테고리의 다른 글

[백준][7570번] 줄 세우기  (0) 2024.02.21
[백준][18429번] 근손실  (0) 2024.01.16
[백준][1138번] 한 줄로 서기  (1) 2024.01.09
[백준][9655번] 돌 게임  (1) 2024.01.08
[백준][1515번] 수 이어 쓰기  (1) 2024.01.04

댓글