[백준 11399] ATM (C++/Python)

C++)

// 11399
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

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

	int n, temp;
	cin >> n;

	vector<int> v;
	for (int idx = 0; idx < n; idx++) {
		cin >> temp;
		v.push_back(temp);
	}

	sort(v.begin(), v.end());

	int ans = 0;
	for (int idx = 0; idx < n; idx++)
		ans += ((n - idx) * v.at(idx));

	cout << ans;
	
	return 0;
}

 

Python)

import sys
read = sys.stdin.readline

n = int(read().rstrip())
arr = list(map(int, read().rstrip().split()))

arr.sort()

ans = 0
for idx, waiting_time in enumerate(arr):
    ans += (n-idx)*waiting_time
print(ans)

 

 

사람의 수가 1,000이므로 정렬해도 1초가 넘지 않는다.