[프로그래머스] 저울

문제 설명

하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 합니다. 이 저울의 양팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같습니다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있습니다.

저울추가 담긴 배열 weight가 매개변수로 주어질 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 return 하도록 solution 함수를 작성해주세요.

예를 들어, 무게가 각각 [3, 1, 6, 2, 7, 30, 1]인 7개의 저울추를 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21입니다.

제한 사항

• 저울추의 개수는 1개 이상 10,000개 이하입니다.
• 각 추의 무게는 1 이상 1,000,000 이하입니다.

입출력 예

weightreturn

[3, 1, 6, 2, 7, 30, 1]

21

입출력 예 설명

문제에 나온 예와 같습니다.

---

어려운 문제였다.

dfs를 이용한 완전 탐색으로 해결하려 했으나 시간 초과가 떴다.

그리디는 규칙을 찾는 게 참 중요하다.

answer와 weight[i]를 더하면서 weight[i]가 더 크다면 answer부터 weight[i]까진 추로 구할 수 없다는 뜻이므로 answer를반환한다. 

처음 추의 무게가 1보다 크다면 1이 답이 되는 것이다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/Programmers/%EC%A0%80%EC%9A%B8.cpp

chosh95/STUDY

C++ 코드

#include <string>
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

int solution(vector<int> weight) {
	int answer = 1;
	sort(weight.begin(), weight.end());
	for (int i = 0; i < weight.size(); i++) {
		if (answer < weight[i]) break;
		answer += weight[i];
	}

	return answer;
}

int main()
{
	cout << solution({ 3,1,6,2,7,30,1 });
}