본문 바로가기
개인 공부/코딩테스트

[C++][백준 11399][그리디 알고리즘] ATM :: seoftware

by seowit 2020. 2. 26.

문제

https://www.acmicpc.net/problem/11399

 

11399번: ATM

첫째 줄에 사람의 수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어진다. (1 ≤ Pi ≤ 1,000)

www.acmicpc.net

소스코드

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
int main(void) {
	int n;
	cin >> n;
	vector<int>line;

	for (int i = 0; i < n; i++) {
		int k;
		cin >> k;
		line.push_back(k);
	}
	sort(line.begin(), line.end());

	int answer = 0;
	for (int i = 0; i < n; i++) {
		answer += line[i] * (n-i);
	}
	cout << answer;

}

1. 입력받은 값을 sort

2. 예를 들어, sort 후에 {1, 2, 3, 4, 5}일 때, 1은 5번 더해지고 2는 4번, 3은 3번, 4는 2번, 5는 1번 더해진다. 그걸 이용해서 line[i] * (n-i) 값을 모두 더해준다. 

댓글