[프로그래머스] LEVEL 2 - 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 = [3, 0, 6, 1, 5]를 기준으로 설명하자면 가장 큰 수인 6에다가 1을 더하여 값이 0으로 이루어진 count 리스트를 만든다.
count = [0,0,0,0,0,0,0]
다음으로 citations를 오름차순 정렬한 후

이 count의 인덱스 보다 citations값이 크거나 같을 경우 count +=1 을 넣어준다. 그럴 경우

count = [5, 4, 3, 3, 2, 2, 1]이 된다.

마지막으로 count의 값이 count 인덱스 값보다 크거나 또는 같을 경우 answer를 갱신해준다.

그럴 경우 인덱스 3에 해당하는 값이 마지막으로 갱신 되어 answer = 3이 된다.

• 코드

def solution(citations):
    citations.sort()
    count = [0] * (max(citations) + 1)
    
    answer = 0
    
    for i in range(len(count)):
        for j in range(len(citations)):
            if i <= citations[j]:
                count[i] += 1
    n = 0
    for i in range(len(count)):
        if count[i] >= n:
            n += 1
            answer = i
    return answer

다른 분들의 코드를 봤는데 정말 간단하게 쓰신 경우가 많아서 스스로 부족함을 많이 느꼈다. 그 중 앞도적인 것은 enumerate를 활용해서 코드를 한줄로 만들어버렸다. 더 공부해야겠다!!!