[11399/JAVA] ATM

• 입력: 기다리는 사람 수 N, N개의 은행업무가 걸리는 시간 Ai
• 출력: 최소 시간
• 알고리즘: 그리디(탐욕법)
• 소스코드1(우선순위 큐 이용)
import java.util.*; public class Main { public static void main(String[] args) { Scanner sc = new Scanner(System.in); int n; PriorityQueue<Integer> wait = new PriorityQueue<Integer>(); for(int i=0; i<n; i++){ int a = sc.nextInt(); wait.add(a); } int total = 0, sub = 0; while(!wait.isEmpty()){ int min = wait.poll(); total += (sub + min); sub += min; } System.out.println(total); } }
• 소스코드2(Arrays sort 이용)
import java.util.*; public class Main { public static void main(String[] args) { Scanner sc = new Scanner(System.in); int n; n = sc.nextInt(); int[] wait = new int[n]; for(int i=0; i<n; i++){ wait[i] = sc.nextInt(); } Arrays.sort(wait); int total = 0, sub = 0; for(int i=0; i<n; i++){ int min = wait[i]; total += (sub + min); sub += min; } System.out.println(total); } }

시간은 우선순위 큐가 조금 더 빨랐다.