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

입출력 예

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

입출력 예 설명

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

1. https://en.wikipedia.org/wiki/H-index "위키백과"

---

나의 코드

1차

def solution(citations):
    citations.sort(reverse=True)
    flag = True
    print(citations)
    for idx, i in enumerate(citations):
        answer = idx
        if i <= idx:
            answer = idx
            flag = False
            break
    if flag == True:
        answer = len(citations)
    return answer

내림차순 정렬 후 인용수가 idx보다 작거나 같아지는 시점에서 값을 구한다.

모든 논문이 길이를 초과해 인용되었다면 제일 큰 idx 즉 배열의 길이가 답이된다.

 

2차 (코드정리)

def solution(citations):
    citations.sort(reverse=True)
    for idx, i in enumerate(citations):
        if i <= idx:
            return idx
            break
    return len(citations)

중간에 끝내도 되는경우 바로 return하면 코드를 줄일 수 있다.

 

프로그래머스 대장답

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

천재적인 발상인듯,,ㅎㅎ

 

참고

어렵지 않다고 느꼈지만, 문제를 이해하는데 오래걸렸다...

아래 링크도 참고하면 좋을 거 같다.

추천 참고링크 : H-Index 설명