[BaekJoon] 11399번 - ATM [Java][C++]

문제 링크

---

1. 문제 풀이

ATM에서 각 사람은 앞사람이 인출하는데 걸리는 시간만큼 기다린 후 인출을 할 수 있다. 이때 첫 번째 사람부터 마지막 사람까지는 첫 번째 사람이 인출하는데 걸리는 시간만큼 기다려야 하고(첫 번째 사람은 자신의 인출 시간만큼 소요), 두 번째 사람부터 마지막 사람까지는 두 번째 사람이 인출하는데 걸리는 시간만큼 또 기다려야 한다. 즉 인출 시간이 짧은 사람부터 먼저 인출해야 전체 인출 시간의 합이 최소가 된다.

---

2. 코드

1. 풀이 [Java]

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int N = Integer.parseInt(br.readLine());

        int[] arr = new int[N];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(arr);

        int sum = 0;
        for (int i = 0; i < N; i++) {
            sum += arr[i] * (N - i);
        }

        System.out.println(sum);
    }
}

2. 풀이 [C++]

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n;
    cin >> n;

    vector<int> v(n);
    for (int& x : v) cin >> x;
    sort(v.begin(), v.end());

    int sum = 0;
    for (int i = 0; i < n; i++) {
        sum += v[i] * (n - i);
    }

    cout << sum;
}

---