프로그래머스 택배 배달과 수거하기 (JAVA)

https://school.programmers.co.kr/learn/courses/30/lessons/150369

프로그래머스

 

저는 구현을 통해 문제를 해결하였습니다.

수거해야하는 것과 배달해야하는 것 중에 긴 것을 찾고 그 거리의 두배를 결과값으로 추가하였습니다.

둘 중 더 먼 곳까지 가면서 배달을 먼저하고 그 다음 수거를 하면 되기 때문에 위와 같은 방식으로 문제를 해결하였습니다.

 

class Solution {
    private static final int NOT_FOUND = -1;
    public long solution(int cap, int n, int[] deliveries, int[] pickups) {
        long answer = 0;
        int delIndex = NOT_FOUND;
        int pickIndex = NOT_FOUND;
        for(int i = n - 1; i >= 0 ; i--){
            if(deliveries[i] > 0 && delIndex == NOT_FOUND){
                delIndex = i;
            }
            if(pickups[i] > 0 && pickIndex == NOT_FOUND){
                pickIndex = i;
            }
        }
        while(delIndex > -1 || pickIndex > -1){
            answer += 2 * Math.max(delIndex+1,pickIndex+1);
            int cnt = 0;
            while(delIndex > -1 && cnt < cap){
                deliveries[delIndex]--;
                cnt++;
                while(delIndex > -1 && deliveries[delIndex] == 0){
                    delIndex--;
                }
            }
            cnt = 0;
            while(pickIndex > -1 && cnt < cap){
                pickups[pickIndex]--;
                cnt++;
                while(pickIndex > -1 && pickups[pickIndex] == 0){
                    pickIndex--;
                }
            }
        }
        return answer;
    }
}