[Baekjoon/C++] 2217번 - 로프

[백준] Baekjoon Online Judge

문제로 이동

 

문제

N(1 ≤ N ≤ 100,000)개의 로프가 있다. 이 로프를 이용하여 이런 저런 물체를 들어올릴 수 있다. 각각의 로프는 그 굵기나 길이가 다르기 때문에 들 수 있는 물체의 중량이 서로 다를 수도 있다.

하지만 여러 개의 로프를 병렬로 연결하면 각각의 로프에 걸리는 중량을 나눌 수 있다. k개의 로프를 사용하여 중량이 w인 물체를 들어올릴 때, 각각의 로프에는 모두 고르게 w/k 만큼의 중량이 걸리게 된다.

각 로프들에 대한 정보가 주어졌을 때, 이 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구해내는 프로그램을 작성하시오. 모든 로프를 사용해야 할 필요는 없으며, 임의로 몇 개의 로프를 골라서 사용해도 된다.

 

입력

첫째 줄에 정수 N이 주어진다. 다음 N개의 줄에는 각 로프가 버틸 수 있는 최대 중량이 주어진다. 이 값은 10,000을 넘지 않는 자연수이다.

 

출력

첫째 줄에 답을 출력한다.

 

---

예제 입력	예제 출력
2 10 15	20

---

풀이

#include <iostream>
#include <set>
using namespace std;

int findMax(int N);

set<int> rope; // 로프의 종류
int arr[10001] = { 0 }; // 각 로프의 개수

// 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구하기
int findMax(int N) {
	int max = 0; // 최대 중량
	int noUse = 0; // 사용하지 않는 로프의 수

	for (set<int>::iterator iter = rope.begin(); iter != rope.end(); iter++) {
		int weight = (N - noUse) * (*iter);

		if (weight > max) max = weight;
		noUse += arr[*iter];
	}

	return max;
}

int main() {
	int N;
	scanf("%d", &N);

	for (int i = 0; i < N; i++) {
		int num;
		scanf("%d", &num);

		arr[num]++;
		rope.insert(num);
	}

	printf("%d\n", findMax(N));

	return 0;
}

일단 각 길이별 로프의 수를 arr[]에 저장했다. 그리고 어떤 길이의 로프들을 가지고 있는지를 중복이 허용되지 않는 set을 사용해서 저장했다.

 

// 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구하기
int findMax(int N) {
	int max = 0; // 최대 중량
	int noUse = 0; // 사용하지 않는 로프의 수

	for (set<int>::iterator iter = rope.begin(); iter != rope.end(); iter++) {
		int weight = (N - noUse) * (*iter);

		if (weight > max) max = weight;
		noUse += arr[*iter];
	}

	return max;
}

set은 자동으로 가지고 있는 값들을 정렬하기 때문에 그냥 순서대로 값을 꺼내서 사용했다. 가장 짧은 로프부터 최대 중량을 계산해서 가장 큰 최대 중량(max)를 구했다. 여기서 noUse는 현재 최대 중량을 구하는 로프보다 짧은 로프의 수를 의미한다. 모든 반복이 끝나면 최대 중량을 return하게 된다.

 

---

2024-11-05 코드 수정

#include <iostream>
using namespace std;

int rope[10001];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    int N;
    
    cin >> N;
    for (int i = 0; i < N; i++) {
        int num;
        cin >> num;
        
        rope[num]++;
    }
    
    int result = 0;
    for (int i = 0; i < 10001; i++) {
        result = max(result, N * i);
        
        N -= rope[i];
    }
        
    cout << result << '\n';

    return 0;
}

 로프의 길이를 정렬하여 사용하기 위해 set을 사용하지 않아도 된다. set을 사용하면 오히려 시간이 더 걸리고, 각 로프의 길이에 대한 수량을 저장하는 배열이 있다면 이를 사용하면 되기 때문이다.