[프로그래머스] 정렬 | H-Index | Level 2 | 파이썬(Python)

letthem·2025년 3월 5일

CodingTest

목록 보기
18/24
post-thumbnail

문제

문제 설명

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

입출력 예

citationsreturn
[3, 0, 6, 1, 5]3

입출력 예 설명

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

풀이

def solution(citations):
    # citations : 논문의 인용 횟수
    # i 점점 커지면서, i번 이상 인용된 것의 개수를 count 하고 끝까지 돌았을 때 같다면 그 i가 정답
    
    answer = 0
    
    count = 0
    
    citations.sort()
    
    arr = []
    
    for i in range(len(citations)):
        for j in citations:
            if citations[j] >= i:
                count += 1
            arr.append(count)
        count = 0
        
    print(arr)
    ...

처음엔 이런식으로 생각하다가 잘 안 돼서
구글링을 해보니 sort를 reverse로 해서 거꾸로 내려오게 하면 더욱 직관적으로 되는 것 같다. (그럼 바로 개수가 세어지니까)

최종 코드

def solution(citations):
    answer = 0
    
    citations.sort(reverse=True) # 내림차순 정렬
    
    for i in range(len(citations)):
        if citations[i] < i + 1: # i + 1 보다 작으면
            return i # i값(n - 1)이 정답

    return len(citations)  # 그 외 : ex) 인용 횟수가 모두 같을 때 : [5, 5, 5, 5] 면 4 이상이 4개라 h-index 4 나와야 함

0개의 댓글