[프로그래머스] Level 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회 이하입니다.

입출력 예

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

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

 

 

제출한 코드

def solution(citations):
    h = 0
    citations.sort()
    for i in range(len(citations)):
        if (len(citations[i:]) > h):
            if len(citations[i:]) >= citations[i]:
                h = citations[i]
            else:
                h = len(citations[i:])
    
    answer = h
    return answer

 

문제 의미가 처음에 약간 헷갈렸는데, 살펴보면 h값의 최대는 주어진 배열의 길이이고,

h값의 최소는 문제 제한사항에 나와있듯 인용횟수가 0회~10000회 이하이므로 0이다.

 

배열을 먼저 정렬해두면 문제 조건에서 나머지 논문이 h번 이하 인용되어야 한다는 조건을 까다롭게 고려할 필요가 없다. 

len(citations[i:]) >= citations[i] 비교를 하는 이유는 예시로 [5,5,5,5] 같은 입력을 고려하기 위함이었다.