[11399] ATM (C++)

#include <iostream>
#include <vector>
#include <numeric>
#include <algorithm>

using namespace std;

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

	int n; cin >> n;
	int sum = 0;
	vector<int> A(n);

	for (int i = 0; i < n; i++)
		cin >> A[i];

	sort(A.begin(), A.end(), greater<int>());

	for (int i = 0; i < n; i++)
		sum += accumulate(A.begin() + i, A.end(), 0);

	cout << sum << endl;

	return 0;
}

 

1. 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 구하려면 시간이 적게 걸리는 순으로 나열

=> 1번은 1번 대기시간, 2번은 (1번+2번)대기시간을 소요하므로, 가장 많이 겹치는 대기시간이 최소가 되어야 함

2. 편의상 내림차순으로 정렬하고 accumulate를 이용해 0~4, 1~4, 2~4, 3~4, 4 index의 합을 모두 sum에 더한 후 출력

(for문의 i 시작점을 익숙한 0으로 놓기 위해 내림차순으로 정렬하였음)