ATM

문제

ATM

풀이

인출하는 데 오래 걸리는 사람 와 빨리 걸리는 사람 가 있습니다. 만약 가 보다 먼저 인출한다면 시간의 합은 가 됩니다. 만약 가 보다 먼저 인출한다면 시간의 합은 가 됩니다. 이므로 빨리 걸리는 사람이 먼저 인출하는 것이 이득입니다.

사람이 세 명 이상이어도 비슷한 원리로 증명할 수 있습니다. 명의 사람이 있을 때, 가 번째로 인출한다면 답에 가 더해집니다. 가 번째로 인출한다면 답에 가 더해집니다. 따라서 가 큰 사람이 작은 사람보다 먼저 인출하는 경우 작은 사람과 인출하는 순서를 서로 바꿨을 때 인출하는데 필요한 총 시간의 합이 무조건 줄어들기 때문에 가 작은 사람부터 인출하는 것이 이득입니다.

코드

Python

n = input()
p = sorted(map(int, input().split()))
 
for i in range(1, n):
    p[i] += p[i - 1]
 
print(sum(p))

C++

#include <bits/stdc++.h>
using namespace std;
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int n;
    cin >> n;
 
    vector<int> p(n);
    for (int i = 0; i < n; i++)
        cin >> p[i];
    
    ranges::sort(p);
    int answer = p[0];
    for (int i = 1; i < n; i++) {
        p[i] += p[i - 1];
        answer += p[i];
    }
 
    cout << answer;
 
    return 0;
}

Java

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.StringTokenizer;
 
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        int n = Integer.parseInt(br.readLine());
        StringTokenizer tk = new StringTokenizer(br.readLine());
        int p[] = new int[n];
        for (int i = 0; i < n; i++)
            p[i] = Integer.parseInt(tk.nextToken());
        
        Arrays.sort(p);
        int answer = p[0];
        for (int i = 1; i < n; i++) {
            p[i] += p[i-1];
            answer += p[i];
        }
 
        bw.write(answer + "");
        bw.close();
    }
}