택배 배달과 수거하기
당신은 일렬로 나열된 n
개의 집에 택배를 배달하려 합니다. 배달할 물건은 모두 크기가 같은 재활용 택배 상자에 담아 배달하며, 배달을 다니면서 빈 재활용 택배 상자들을 수거하려 합니다.
배달할 택배들은 모두 재활용 택배 상자에 담겨서 물류창고에 보관되어 있고, i
번째 집은 물류창고에서 거리 i
만큼 떨어져 있습니다. 또한 i
번째 집은 j
번째 집과 거리 j - i
만큼 떨어져 있습니다. (1 ≤ i
≤ j
≤ n
)
트럭에는 재활용 택배 상자를 최대 cap
개 실을 수 있습니다. 트럭은 배달할 재활용 택배 상자들을 실어 물류창고에서 출발해 각 집에 배달하면서, 빈 재활용 택배 상자들을 수거해 물류창고에 내립니다. 각 집마다 배달할 재활용 택배 상자의 개수와 수거할 빈 재활용 택배 상자의 개수를 알고 있을 때, 트럭 하나로 모든 배달과 수거를 마치고 물류창고까지 돌아올 수 있는 최소 이동 거리를 구하려 합니다. 각 집에 배달 및 수거할 때, 원하는 개수만큼 택배를 배달 및 수거할 수 있습니다.
https://school.programmers.co.kr/learn/courses/30/lessons/150369
import java.util.Arrays;
public class Main {
public static void main(String[] args) {
int cap = 4;
int n = 5;
int[] deliveries = {1, 0, 3, 1, 2};
int[] pickups = {0, 3, 0, 4, 0};
var reuslt = solution(cap, n, deliveries, pickups);
System.out.println(reuslt);
}
public static long solution(int cap, int n, int[] deliveries, int[] pickups) {
long result = 0;
int check_D = deliveries.length - 1;
int check_P = pickups.length - 1;
int totalDeliveries = Arrays.stream(deliveries).sum();
int totalPickUps = Arrays.stream(pickups).sum();
int scDeliverires = 0;
int scPickups = 0;
while (totalDeliveries != scDeliverires || totalPickUps != scPickups) {
int delvBox = 0;
int go = 0;
boolean last = false;
for (int i = check_D; i > -1; i--) {
if (deliveries[i] == 0) {
continue;
}
if (!last) {
go = i + 1;
last = true;
}
while (delvBox < cap && deliveries[i] != 0) {
delvBox++;
scDeliverires++;
deliveries[i]--;
}
if (delvBox == cap) {
check_D = i;
break;
}
}
last = false;
int pickUpBox = 0;
int pickupGo = 0;
for (int i = check_P; i > -1; i--) {
if (pickups[i] == 0) {
continue;
}
if (!last) {
pickupGo = i + 1;
last = true;
}
while (pickUpBox < cap && pickups[i] != 0) {
pickUpBox++;
scPickups++;
pickups[i]--;
}
if (pickUpBox == cap) {
check_P = i;
break;
}
}
if (pickupGo > go) {
result += pickupGo * 2;
} else {
result += go * 2;
}
}
return result;
}
}