[프로그래머스/Python] 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회 이하입니다.

입출력 예

입출력 예 설명

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

---

풀이

def solution(citations):
    answer = 0
    citations.sort()
    
    for i in citations:
        count = 0
        for j in citations:
            if j >= i:
                count += 1
        if i >= count:
            if answer < count:
                answer = count
    return answer

 

이해하기가 까다로웠던 문제.

문제 내용 중 'h번 이상 인용된 논문이 h편 이상'을 집중해서 생각했다.

 

[풀이 과정]

1. 주어진 값 정렬

2. citations의 값들을 반복문을 사용하여 기준 값(i)보다 같거나 큰 값을 count

3. 기준 값(i)이 count값보다 같거나 클 때, 가장 큰 값을 출력

 

ex 1) [3, 0, 6, 1, 5]일 경우

i	count
0	5
1	4
3	3
5	2
6	1

 

i가 count보다 같거나 클 경우는 i가 3, 5, 6일 경우이고, 그 경우에서 가장 큰 count를 구한다.

 

ex 2) [6, 5, 5, 5, 3, 2, 1, 0]일 경우

i	count
0	8
1	7
2	6
3	5
5	4
5	4
5	4
6	1

 

i가 count보다 같거나 클 경우는 i가 5, 6일 경우이고, 그 경우에서 가장 큰 count인 4가 기대값.

---

[기존 코드]

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

 

기존에는 answer을 배열로 만들어서 count값을 넣고 가장 큰 값을 답으로 작성했으나 런타임 에러로 정답 코드로 변경~

참고🐣

[참고 코드]

def solution(citations):
    citations.sort(reverse=True)
    answer = max(map(min, enumerate(citations, start=1)))
    return answer

enumerate() 함수는 기본적으로 인덱스와 원소로 이루어진 튜플(tuple)을 만들어 줌

start=1은 첫 인덱스를 1로 바꿔준다

 

- 해당 코드 풀이 과정

1. sort로 정렬해서 가장 큰값부터 작은값으로 정렬하고 enumerate로 (index, value)형태로 묶는다.

2. 최댓값(start = 1)부터 각 value에 대해 최솟값 value의 값을 min으로 추출하고, 이 추출된 값은 enumerate가 끝나는 citations 리스트의 크기에 해당하는 갯수가 나온다.

3. 이들을 map으로 묶으면, 한 value의 입장에서 보는 최솟값 value의 집합이 나온다. 즉 h값들의 집합이나온다.

4. h값중 최대값을 max로 뽑아서 출력

 

문제 출처

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