[백준 / JAVA] 11399번, ATM (Greedy)

---

풀이

생각한 접근법은 ‘대기 시간의 총합’을 구하는 것이기에 그리디 알고리즘을 생각했다.

1. 사람들을 인출 시간이 짧은 순서로 정렬하면, 이후 사람들의 대기 시간을 최소화 할 수 있다.
2. 최대 사람 수가 1000명으로 정해 졌기에 Counting Sort를 사용하는게 용이하다.
3. → 등장 횟수를 누적합으로 구하기 쉬웠기 때문에.

해당 부분에 대한 코드는 밑 처럼 될 것이며.

        int total_time = 0;
        int prev_time = 0;

        for (int i = 1; i < Max; i++) {
            while(person[i] --> 0){
                total_time += (i + prev_time);
                prev_time += i;
            }
        }

 

전체코드는 아래처럼 될 것이다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    final static int Max = 1001;
    static int[] person;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());

        person = new int[Max];
        st = new StringTokenizer(br.readLine());

        // Counting sort!!!
        while(n --> 0){
            person[Integer.parseInt(st.nextToken())]++;
        }

        int total_time = 0;
        int prev_time = 0;

        for (int i = 1; i < Max; i++) {
            while(person[i] --> 0){
                total_time += (i + prev_time);
                prev_time += i;
            }
        }
        System.out.println(total_time);
    }
}