프로그래밍/Python
[프로그래머스] H-Index / 정렬 / python
나는 라미
2024. 3. 16. 21:53
728x90
반응형
문제
https://school.programmers.co.kr/learn/courses/30/lessons/42747
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회 이하입니다.
풀이
아이디어
보자마자 자연스레 이분탐색이 떠올랐다
return은 h번 이상 인용된 논문의 개수 : mid가 되겠다
최대가 될 수 있는 값은 가장 많이 인용된 논문의 수가 된다
: end = max(citations)
이분탐색 후 최대가 되는 h값을 리턴해주면 끝~
코드
def solution(citations):
start, end = 0, max(citations)
answer = 0
while start <= end:
mid = (start + end) // 2
temp = [i for i in citations if i >= mid]
if len(temp) < mid:
end = mid -1
else:
answer = max(mid,answer)
start = mid +1
return answer728x90
반응형