택배 배달과 수거하기

택배 배달과 수거하기

재할용 택배 상자.png

당신은 일렬로 나열된 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;
    }
}