[프로그래머스 Lv.2] H-Index (Python)

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42747

프로그래머스

풀이

정렬한 인용 횟수의 배열과 인덱스와 전체 길이를 사용해서 문제를 해결할 수 있다.

def solution(citations):
    citations.sort() # 배열 정렬
    idx = 0
    length = len(citations)
    
    if citations[idx] > length: # 인용 횟수의 최소값이 이미 전체 논문 수를 초과한다면
        return length # H-Index는 논문 수와 동일
    
    
    for i in range(length):
        if length - i >= citations[i]: # 현재 논문의 인용수보다 더 큰 인용수를 가지는 논문의 수가 현재 논문의 인용수 이상이라면
            idx = i # 인덱스값을 저장
    
    answer = citations[idx]
    if idx != length - 1: # 우선 현재 인덱스가 최대 인덱스인지 체크, 최대 인용수는 최대 H-Index임
        for j in range(answer + 1, citations[idx+1]): # 현재 논문의 인용수보다 H-Index가 더 큰 경우를 체크하기 위함
            if length - idx - 1 < j: # 현재 인용수보다 큰 인용수를 가지는 논문의 수(length-i-1)가 현재 인용수보다 적다면 break
                break
            answer = j
    return answer

애초에 index를 기준으로 for문을 순회하는 것이 아니라 인용수를 기준으로 잡고 순회한다면 더 깔끔한 코드가 나올 것 같아 최종적으로는 아래와 같이 풀이하였다.

def solution(citations):
    citations.sort()
    length = len(citations)
    
    for i in range(length):
        if citations[i] >= length-i: # 현재 인용 수 이상 인용된 논문의 수가 현재 인용 수보다 작다면
            return length-i # h번 이상 인용된 논문의 수를 반환
    return 0 # 모든 논문의 인용 수가 0인 경우

훨씬 깔끔해진 코드를 확인할 수 있다.