[BOJ][Python][C++] 백준 16208번 - 귀찮음

문제 링크: https://www.acmicpc.net/problem/16208

16208번: 귀찮음

---

문제 풀이

쉽고 쉬운 그리디 문제. xy의 값이 작을수록 값이 작아지기 때문에 막대기의 총합을 구하고 정렬을 한 후 첫번째 막대부터 차례로 총합에서 빼고 곱해주면 된다. 즉 x가 막대 1개, y가 x를 제외한 합이라고 생각하면 된다.

 

코드

	n = int(input())
	steel = sorted(list(map(int, input().split())))
	ans = 0
	s = sum(steel)
	for i in range(n):
	s -= steel[i]
	ans += steel[i] * s
	print(ans)

view raw 16208.py hosted with ❤ by GitHub

파이썬 코드

 

 

	#include <iostream>
	#include <algorithm>
	#include <vector>
	using namespace std;
	int main() {
	long long n, x, sum = 0, ans = 0;
	vector<int> steel;
	cin >> n;
	for (int i = 0; i < n; i++){
	cin >> x;
	sum += x;
	steel.push_back(x);
	}
	for (int i = 0; i < n; i++){
	sum -= steel[i];
	ans += steel[i]*sum;
	}
	cout << ans << '\n';
	}

view raw 16208.cpp hosted with ❤ by GitHub

C++ 코드