programmers_택배 배달과 수거하기_java

2023. 6. 1. 15:52Algorithm/Programmers

728x90

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

class Solution {
    public long solution(int cap, int n, int[] deliveries, int[] pickups) {
        long answer = 0;

        int give = 0; // 배달
        int get = 0; // 수거

        for (int i = n; i>0; i--){
            if (deliveries[i-1] != 0 || pickups[i-1] != 0){
                int cnt = 0; // 해당 장소 방문 필요 횟수
                // 해당 장소에 방문 필요 횟수 구하기
                while(give < deliveries[i-1] || get < pickups[i-1]){ 
                    cnt++;
                    give += cap;
                    get += cap;
                }
                // 방문 하고 남은 것
                give -= deliveries[i-1];
                get -=  pickups[i-1];
                
                answer += (long)(i*cnt*2); // 거리 * 반복횟수 * 왕복
            }
        }

        return answer;
    }
}