[프로그래머스] level2. H-Index

문제 설명

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과1에 따르면, H-Index는 다음과 같이 구합니다.

어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다.

어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

 

나의 풀이

def solution(citations):
    answer = 0
    temp = 0
    for h in range(1, max(citations)):
        for i in citations:
            if i >= h:
                temp += 1
        if temp >= h:
            answer = h
        temp = 0
        
    return answer

다른 사람들의 풀이를 보다가 너무 똑똑하게 풀어서 깜짝 놀란 풀이가 있는데요, 이는 다음과 같습니다.

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

 

citation을 내림차순으로 정렬한 뒤, enumerate를 1부터 지정해주면 해당 인덱스 값이 h번 이상 인용된 논문의 수를 가리키게 됩니다. 이를 map을 통해 min(h번 이상 인용된 논문의 수, h)를 한 뒤, max를 통해 가장 큰 값을 구하면 답을 구할 수 있다.