프로그래머스/Lv.2

[프로그래머스/Python] H-Index

KimNang 2023. 1. 30. 11:58

문제 정보

제목 : H-Index

난이도 : Lv.2

사용 언어 : Python

문제 링크

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

 

문제 설명

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과에 따르면, 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입니다.


나의 풀이

문제 설명으로는 이해가 잘 되지않아 문제에 있는 링크를 통해 H-Index에 대해 자세히 알아보았습니다. 그 후 설명을 보고 설명 그대로 코드를 짰습니다. 비효율적인 코드로 짜게 되어서 아쉬움이 많은 문제였습니다.

먼저 논문을 내림차순으로 정렬합니다. 이후 citations의 길이만큼 반복하면서 i+1이 i번째의 인용 횟수랑 같으면 i+1을, 작으면 i를, for문이 끝났지만 결과를 얻지 못하면 i+1을 answer에 저장하여 리턴합니다.

 

코드

def solution(citations):
    answer = 0
    citations.sort(reverse=True)
    
    for i in range(len(citations)) :
        if (i + 1 == citations[i]) :
            answer = i+1
            break
        elif (i + 1 > citations[i]) :
            answer = i
            break
        elif ( i == len(citations)-1) :
            answer = i+1

    return answer