Programmers, 연습문제, 정렬, ‘H-Index’

프로그래머스 연습문제 중 42747번 ‘H-Index’

https://programmers.co.kr/learn/courses/30/lessons/42747

문제 설명

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과1에 따르면, H-Index는 다음과 같이 구합니다. 어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다. 어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 과학자가 발표한 논문의 수는 1편 이상 1,000편 이하입니다.
• 논문별 인용 횟수는 0회 이상 10,000회 이하입니다.

입출력 예

|citations|return| |-|-| |[3, 0, 6, 1, 5]|3|

입출력 예에 대한 설명

이 과학자가 발표한 논문의 수는 5편이고, 그중 3편의 논문은 3회 이상 인용되었습니다. 그리고 나머지 2편의 논문은 3회 이하 인용되었기 때문에 이 과학자의 H-Index는 3입니다.

---

풀이

간단하지만 생각하기가 약간 까다로운 문제였다.

먼저 주어진 논문의 인용횟수 리스트 citaitons를 인용횟수 오름차순으로 정렬한다. 그리고 각 논문의 인용횟수에 대한 h_index를 구하고 그 중 최댓값을 구하면 된다.

i번째 원소의 h_index는 h번 이상 인용된 논문이 h편 이상있는 경우의 h이므로 citations[i]와 n - i (즉 citations[i]이상의 인용횟수를 가진 논문의 수, 정렬된 상태이므로 가능하다.)중 더 작은 값이 곧 h가 된다.

이를 코드로 나타내면 다음과 같다.

def solution(citations):
    citations.sort()
    n = len(citations)
    h = 0
    for i, citation in enumerate(citations):
        h = max(h, min(n-i, citation))
    return h