<청춘> 격정적으로 사는 것

밤을 새고 공부한 다음 날 새벽에 느꼈던 생생한 환희와 야생적인 즐거움을 잊을 수 없다

코딩테스트

[그리디 Greedy] 01. 모험가 길드

수학도 2021. 8. 10. 01:00

그리디 (Greedy)

현재 상황에서 가장 좋아 보이는 것만을 선택하는 알고리즘

 


 

모험가 길드

난이도 ★☆☆  풀이 시간 30분  시간 제한 1초

 

한 마을에 모험가가 N명 있습니다. 모험가 길드에서는 N명의 모험가를 대상으로 '공포도'를 측정했는데, '공포도'가 높은 모험가는 쉽게 공포를 느껴 위험 상황에서 제대로 대처할 능력이 떨어집니다.

모험가 길드장인 동빈이는 모험가 그룹을 안전하게 구성하고자 공포도가 X인 모험가는 반드시 X명 이상으로 구성한 모험가 그룹에 참여해야 여행을 떠날 수 있도록 규정했습니다. 동빈이는 최대 몇개의 모험가 그룹을 만들 수 있는지 궁금합니다.

동빈이를 위해 N명의 모험가에 대한 정보가 주어졌을 때, 여행을 떠날 수 있는 그룹 수의 최댓값을 구하는 프로그램을 작성하세요.

 

예를 들어 N = 5이고, 각 모험가의 공포도가 다음과 같다고 가정합시다.

2 3 1 2 2

 

이때, 그룹 1에 공포도가 1, 2, 3인 모험가를 한 명씩 넣고, 그룹 2에 공포도가 2인 남은 두 명을 넣게 되면, 총 2개의 그룹을 만들 수 있습니다. 또한 몇 명의 모험가는 마을에 그대로 남아 있어도 되기 때문에, 모든 모험가를 특정한 그룹에 넣을 필요는 없습니다.

 

입력 조건

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

출력 조건

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

 

입력 예시

5
2 3 1 2 2

 

출력 예시

2

 


코드

# 모험가의 수
N = int(input())
fears = list(map(int, input().split()))
fears.sort()

# 총 그룹 수
total_group = 0
# 현재 그룹에 포함된 모험가의 수
now_group = 0

for fear in fears:
    now_group += 1
    # 현재 그룹에 포함된 모험가의 수가 현재의 공포도 이상이라면, 그룹 결성
    if now_group >= fear:
        # 총 그룹 수 증가
        total_group += 1
        # 현재 그룹 모험가 수 초기화
        now_group = 0
        
print(total_group)

해설

여행을 떠날 수 있는 그룹 수의 최댓값을 구하는 프로그램을 작성하세요.

모든 모험가를 특정한 그룹에 넣을 필요는 없습니다.

→ 최댓값을 구해야하므로 오름차순 정렬을 수행한다.

 

이유

예를 들어 4 2 3 1 이 있으면

오름차순 정렬 시 : 1 2 3 4

{1}, {2, 3}과 같이 두개의 그룹을 만들 수 있다.

내림차순 정렬 시: 4 3 2 1

{4, 3, 2, 1}과 같이 한개의 그룹을 만들 수 있다.

 

 


출처

나동빈, 『이것이 취업을 위한 코딩테스트다 with 파이썬』, 한빛미디어(주), 2020년