[Python] Lv2.H-Index

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

코딩테스트 연습 - H-Index

 

 

 

 

 

문제 설명

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

 

나의 접근 방법 및 풀이

주어진 인용 횟수 배열 citations의 최댓값만큼 for() 문을 반복하며 최대 인용 횟수를 찾는다.

citations의 각 요소의 값이 첫 번째 for문의 i값(인용 횟수) 보다 같거나 클 경우 count를 1씩 증가한다.

그 후, '발표한 논문 n 편 중, h번 이상 인용된 논문이 h 편 이상이고 나머지 논문이 h번 이하 인용되었다면'라는 조건처럼

count가 i(인용 횟수) 이상이고, 나머지 논문(clen-count)이 i이하라면 해당 i값을 h_max, 즉 인용 최댓값으로 정의하며 최종적으로 for() 문이 종료되면 해당 h_max값을 반환한다.

def solution(citations):
    h_max = 0   ## 인용 최댓값
    clen = len(citations)   ## 발표한 논문 수
    for i in range(1, max(citations)+1):
        count = 0
        for j in citations:
            if i <= j:
                count += 1
        if count >= i and clen-count <= i:
            h_max = i
        
        
    return h_max

 

다른 사람 풀이

1. citations를 정렬한다.(오름차순으로)

2. citations의 길이를 l에 저장한다.

3. l만큼(citations의 길이만큼) for() 문을 반복하며, citations의 각 요소의 값이 l-i, 즉 citations의 길이에서 해당 요소의 인덱스 값을 뺀 값보다 크거나 같을 경우 그 값을 바로 반환한다.

def solution(citations):
    citations = sorted(citations)		##[0, 1, 3, 5, 6]
    l = len(citations)
    for i in range(l):
        if citations[i] >= l-i:
            return l-i
    return 0