프로그래머스 LV.2 - H-Index[Python]

문제 설명

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회 이하입니다.

입출력 예citationsreturn

[3, 0, 6, 1, 5]

3

입출력 예 설명

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

---

문제를 풀기에 앞서,

citations = [6,6,6,6,6,1] 일 때, 이 과학자의 H-index는?

이 질문을 풀어보자. 답은 5이다.
이 예시를 든 이유는 일부 사람들이 h 값을 citaions 리스트 안에 있는 값 중에서 고르려고 하기 때문이다. 
즉, 우리는 H-index의 최댓값과 리스트의 길이의 관계를 고려해서 풀어야 한다.

---

나의 풀이

def solution(citations):
    citations.sort()
    article_count = len(citations) # 5

    for i in range(article_count): #0~4
        if citations[i] >= article_count-i:
            return article_count-i
    return 0

• if citations[i] >= article_count-i는 "주어진 h번 이상 인용된 논문이 h편 이상"이라는 조건을 그대로 풀어쓴 것이었다.
• citations[i]는 i번 논문이 인용된 횟수이고 article_count - i는 인용된 논문의 개수를 최댓값부터 하나씩 줄여나간 것이다. (최댓값을 찾아야 하므로 가장 큰 값부터 시작)
• 정렬된 citations에서 한칸씩 가다가 그 값이 나머지 길이보다 크거나 같다면 그 편수가 h다.
• 그리고 리스트는 오름차순 정렬된 상태이므로 i번째 이후는 모두 i번째보다 큰 값을 가질 것이다.

 
https://www.ibric.org/myboard/read.php?Board=news&id=270333

---

신박한 풀이

def solution(citations):
    citations.sort(reverse=True)
    answer = max(map(min, enumerate(citations, start=1)))
    return answer