백준 온라인 저지 2437 저울
문제
하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 한다. 이 저울의 양 팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있다.
무게가 양의 정수인 N개의 저울추가 주어질 때, 이 추들을 사용하여 측정할 수 없는 양의 정수 무게 중 최소값을 구하는 프로그램을 작성하시오.
예를 들어, 무게가 각각 3, 1, 6, 2, 7, 30, 1인 7개의 저울추가 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최소값은 21이다.
입력
첫 째 줄에는 저울추의 개수를 나타내는 양의 정수 N이 주어진다. N은 1 이상 1,000 이하이다. 둘째 줄에는 저울추의 무게를 나타내는 N개의 양의 정수가 빈칸을 사이에 두고 주어진다. 각 추의 무게는 1이상 1,000,000 이하이다.
7
3 1 6 2 7 30 1
출력
첫째 줄에 주어진 추들로 측정할 수 없는 양의 정수 무게 중 최소값을 출력한다.
21
백준 온라인 저지
소스코드
#include <iostream>
#include <algorithm>
using namespace std;
int main() {
int N;
cin >> N;
int * arr = new int[N];
int * arrSum = new int[N];
int min = 1;
for (int i = 0; i < N; i++) {
cin >> arr[i];
}
sort(arr, arr + N);
if (arr[0] != 1) {
min = 1;
}
else {
arrSum[0] = arr[0];
for (int i = 1; i < N ; i++) {
arrSum[i] = arrSum[i - 1] + arr[i];
min = arrSum[i] + 1;
if (i == N - 1) {
break;
}
if (min < arr[i + 1])
break;
}
}
cout << min << endl;
return 0;
}
그리디 알고리즘을 사용한다. 각각의 경우가 최적이면 최종해도 최적이다.라는 원칙으로 풀이를 진행한다. 핵심부분은 if(min<arr[i+1])
으로 최소값을 계속해서 갱신하다 조건절이 참이되면 반복문을 멈추고 min값을 출력한다. 지금까지 합한 값보다 그 다음arr
의 값이 크다면 arrSum
과 arr
의 숫자 사이에 공백이생길 것이다. 문제에서 원하는 값은 만들지 못하는 최솟값이므로 반복문으로 멈추고 min
을 출력하도록한다.