[이코테] 그리디 - 모험가 길드

문제

한 마을에 모험가가 N명 있습니다. 모험가 길드에서는 N명의 모험가를 대상으로 '공포도'를 측정했는데, '공포도'가 높은 모험가는 쉽게 공포를 느껴 위험 상황에서 제대로 대처할 능력이 떨어집니다. 모험가 길드장인 동빈이는 모험가 그룹을 안전하게 구성하고자 공포도가 X인 모험가는 반드시 X명 이상으로 구성한 모험가 그룹에 참여해야 여행을 떠날 수 있도록 규정했습니다. 동빈이는 최대 몇개의 모험가 그룹을 만들 수 있는지 궁금합니다. 동빈이를 위해 N명의 모험가에 대한 정보가 주어졌을 때, 여행을 떠날 수 있는 그룹 수의 최댓값을 구하는 프로그램을 작성하세요. (몇 명의 모험가는 마을에 그대로 남아 있어도 되기 때문에, 모든 모험가를 특정한 그룹에 넣을 필요는 없습니다.)

 

입력

• 첫째 줄에 모험가의 수 N이 주어집니다.(1<=N<=100,000)
• 둘째 줄에 각 모험가의 공포도의 값을 N 이하의 자연수로 주어지며, 각 자연수는 공백으로 구분합니다.

출력

• 여행을 떠날 수 있는 그룹 수의 최댓값을 출력합니다.

 

내 코드 - Python

가장 큰 수 n부터 n만큼 채우고 다음 그룹도 똑같은 방식으로 채우기

import sys

n = int(sys.stdin.readline())
arr = list(map(int, sys.stdin.readline().split()))
arr.sort(reverse=True)
result = 0
current = 0

while True:
    current += arr[current]
    result += 1
    # 인덱스 범위를 넘겼을 경우
    if current > n:
        result -= 1
        break
    # 마지막 인덱스일 경우
    elif current == n - 1:
        # 마지막 원소 값이 1일 경우 +1
        if arr[current] == 1:
            result += 1
        break
    elif current == n:
        break

print(result)

 

다른 풀이 코드 - Python

오름차순으로 공포도를 하나식 확인하며, '현재 그룹에 포함된 모험가의 수'가 '현재 확인하고 있는 공포도'보다 크거나 같다면 이를 그룹으로 설정한다.

import sys

n = int(sys.stdin.readline())
arr = list(map(int, sys.stdin.readline().split()))
arr.sort()
result = 0
count = 0

for i in arr:
    count += 1
    if count >= i:
        result += 1
        count = 0

print(result)

 

다른 풀이 코드 - C++

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin.tie(0);
    
    int n, result = 0, count = 0;
    vector<int> arr;

    cin >> n;

    for (int i = 0; i < n; i++) {
        int x;
        cin >> x;
        arr.push_back(x);
    }

    sort(arr.begin(), arr.end());

    for (int i = 0; i < n; i++) {
        count += 1;
        if (count >= arr[i]) {
            result += 1;
            count = 0;
        }
    }

    cout << result << '\n';
}

 

분석

내 코드는 내림차순으로 그룹을 나눠서 예외 처리하느라 코드가 지저분하다 ㅋㅋ..

다른 풀이 코드는 오름차순으로 나눈 것이다.