파이썬(Python) 알고리즘 문제풀이 H-Index [프로그래머스 / 정렬]

출처) 프로그래머스 Facebook

문제 설명

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	result
[3, 0, 6, 1, 6]	3

 

 

입출력 예시 설명

입출력 예 #1

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

 

 

풀이 코드

citations 리스트 길이가 10^3 이하이기 때문에 이중 for 문으로 풀 수도 있지만, 정렬 카테고리에 있는 문제이기 때문에 오름차순 정렬을 활용해 문제를 풀이했다. 이해하면 어렵지 않지만 처음에 문제 이해가 어려워 질문하기 등을 보고 힌트를 얻어 풀이했다. 내림차순으로도 풀 수 있는 것 같고 .. 문제 설명을 어렵게 적어놓은건지 내 문해력이 떨어지는 건지 가늠하기 힘들었음 ;;

def solution(citations):
	citations = sorted(citations)
    
    # [0, 1, 3, 5, 6]
    # 만약에 h 기준이 1이라고 하면, 1번 이상 인용된 논문이 1개 이상인지? 확인하면 된다.
    # 오름차순이 돼있으니까 index 를 구해서 전체 리스트 길이에서 1 이 존재하는 index 길이를 빼면 그 뒤에는 
    # 어짜피 전부 1 이상이기 때문에 1번 이상 인용된 논문의 개수를 파악할 수 있다.
    for idx, citation in enumerate(citations):
    	if citation >= len(citations) - idx:
        	return len(citations) - idx
    
    return 0