알고리즘 풀이: [프로그래머스] H-Index

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

코딩테스트 연습 - H-Index

문제 설명

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
[6, 6, 6, 6, 6]	5

 

입출력 예 설명

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

---

문제 풀이

인용 횟수를 정렬하고, 언제 H Index의 최대값을 가지는지 파악합니다. 정렬된 인용 횟수에서 i 번째 인용 횟수 citations[i]가 그 인용 횟수 이상의 논문 수인 (전체 논문 수 - i) 보다 크거나 같은 지점에서 H Index의 최댓값 (전체 논문 수 - i)가 결정됩니다.

 

예제를 통해 확인합니다.

 

인용 횟수에 따른 최대 H Index 찾기

인용 횟수를 정렬합니다. 따라서 i 번째 인용 횟수 citations[i] 보다 많은 인용 횟수를 가진 논문의 수는 (전체 논문의 수 - i)가 됩니다. 이를 고려하면 H Index의 최댓값은 citations[i]가 (전체 논문의 수 - i) 보다 크거나 같은 시점에서 결정되는 것을 알 수 있습니다.

 

소스 코드

메인 솔루션 함수

def solution(citations):
    # 인용 횟수를 오름차순으로 정렬
    citations = sorted(citations)
    # 전체 논문의 수
    n = len(citations)
    # H Index 초기값은 0이다. 일단 논문이 제출되면 0 번 인용 횟수를 가진 논문은 0개 이상이다.
    h_index = 0
    for idx, citation in enumerate(citations):
        # i 번째 인용 횟수가 남은 논문의 수보다 많은 지점에서 최대 H Index 가 결정된다
        if citation >= (n - idx):
            h_index = (n - idx)
            break

    return h_index


if __name__ == '__main__':
    assert solution([3, 0, 6, 1, 5]) == 3
    assert solution([6, 6, 6, 6, 6]) == 5

 

프로그래머스

github 주소: https://github.com/dhsong95/-PRACTICE-Programmers-Algorithm/blob/master/level%202/H%20Index.py

dhsong95/-PRACTICE-Programmers-Algorithm