[프로그래머스] H-Index (level2, 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회 이하입니다.

---

입출력 예

 

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

---

입출력 예 설명

 

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

---

 

풀이 코드

def solution(citations):
    answer = 0
    citations.sort()
    whole = len(citations)
    
    for h in range(citations[-1] + 1):
        cnt = 0
        for cite in citations:
            if cite < h:
                cnt += 1
            else:
                break
        if whole - cnt >= h and cnt <= h:
            answer = h
        else:
            break
            
    return answer

정렬 후 문제에 맞게 구현하면 풀리는 문제이다.

문제 풀이에 중요한 점은, h값은 citations 배열 내에 들어있지 않을 수도 있다라는 점이다.

따라서 H-Index가 될 수 있는 후보는 정렬한 후에 0~citations[-1]이므로, 이 범위 내에서 문제에 주어진 조건을 구현하면 된다.

이 점을 깨닫고 나면 문제를 풀이하는데 크게 어려운 점은 없지만, 자칫 citations 내에 있다고 판단하기 쉽다.