[BOJ] 11399번 ATM - C++

"그리디 알고리즘"

Posted by Yongmin on June 14, 2021

문제

ATM

풀이

i번째에 해당하는 사람이 총 걸리는 시간은 i-1까지의 시간 + 본인이 걸리는 시간이다. 그러므로 최솟값은 서있는 사람들을 걸리는 시간이 작은 순서대로 줄을 서게 하는 것이 최솟값이 될 것이다. 이는 결국 최솟값은 첫번째 사람은 N번만큼 연산, 두번째 사람은 N-1번만큼 … N번째 사람은 1번만큼 연산되는 것으로 결과 도출이 가능하다.

소스 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main(){
    int N;
    scanf("%d", &N);
    
    vector<int> arr;
    for(int i = 0; i<N; i++){
        int input;
        scanf("%d", &input);
        arr.push_back(input);
    }
    
    sort(arr.begin(), arr.end());
    
    int sum = 0;
    for(int i = 0; i<N; i++){
        sum = sum + arr[i]*(N-i);
    }
    printf("%d\n", sum);
}


# # #