알고리즘/Sort 정렬

[프로그래머스] H-Index - 파이썬(python)

아뵹젼 2022. 9. 6.

문제 설명

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

 

 

나의 풀이

def solution(citations):
    docu = sorted(citations,reverse=True)
    for i in range(len(docu)) :
        if docu[i] <= i :
            return i
    return len(docu)

h번 이상 인용된 논문의 개수가 h편 이상이다.

 

논문인용횟수를 내림차순으로 정렬했다.

h는 0부터 시작하기 때문에, i번째에서 조건을 만족한다면, h를 리턴한다.

(해당 인덱스 앞까지 h개의 논문이 h번 이상 인용된 것)

[6, 5, 5, 5, 3, 2, 1, 0] 를 예시로 들어보자.

6 <= 0 (x)

5 <= 1 (x)

5 <= 2 (x)

5 <= 3 (x)

3 <= 4 (o) -> 실패, 따라서 h는 4.

따라서 h는 4가 된다. 4편이상 인용한 논문의 개수는 4개이다.

 

[6,5,4,1,0] 을 예시로 들어보자.

6 <= 0 (x)

5 <= 1 (x)

4 <= 2 (x)

1 <= 3 (o)

따라서 h는 1이다.

 

 

다른 풀이

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

논문인용횟수를 내림차순으로 정렬했다.

1부터 시작한 인덱스와, 해당 인덱스에 인용된 논문의 개수 둘 중에 더 작은 값이 h가 될 수 있는 값이다.

 

 

댓글