https://www.acmicpc.net/problem/11399
이 문제는 정렬을 이용하는 문제이다. 시간을 기준으로 오름차순 정렬을 한 후, 누적합을 구해 누적합끼리의 합을 구하면 된다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int N = Integer.parseInt(br.readLine()); //사람의 수
String[] input = br.readLine().split(" ");
int[] P = new int[N + 1]; //각 사람이 돈을 인출하는 데 걸리는 시간
int[] dp = new int[N + 1]; //누적합을 저장할 배열
//index : 사람 번호
//P[index] : 그 사람이 걸리는 시간
for (int i = 1; i <= N; i++) {
P[i] = Integer.parseInt(input[i - 1]);
}
//각 사람이 돈을 인출하는 데 시간이 가장 적게 걸려야 함 -> 시간을 기준으로 오름차순 정렬
Arrays.sort(P);
dp[0] = 0;
dp[1] = P[0]; //정렬 후 가장 처음 사람의 대기 시간 저장
for (int i = 1; i <= N; i++) {
dp[i] = dp[i - 1] + P[i]; //이전까지의 누적합 + 현재값
}
int sum = 0;
for (int i = 1; i <= N; i++) {
sum = sum + dp[i];
}
System.out.println(sum);
br.close();
}
}
출력 초과는 처음보는데, print문을 문제에서 주는 것 이외에 작성할 때 발생한다고 한다ㅠㅠ