Sort: H-Index [Lv2]
문제 설명
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 | return |
|---|---|
| [3, 0, 6, 1, 5] | 3 |
입출력 예 설명
이 과학자가 발표한 논문의 수는 5편이고, 그중 3편의 논문은 3회 이상 인용되었습니다. 그리고 나머지 2편의 논문은 3회 이하 인용되었기 때문에 이 과학자의 H-Index는 3입니다.
Solution
내 풀이
def solution(citations):
answer = len(citations)
citations.sort(reverse=True)
for i, v in enumerate(citations):
if v < i+1:
answer = i
break
return answer주어진 citataions를 내림차순으로 정렬 후 for문와 enumerate를 통해 값을 비교한다.
v가 i+1보다 작다면 i번째 까지 값들은 모두 i번째 값보다 크고 그 개수가 i개라는 뜻이다.
예를 들면 citations가 [3, 0, 6, 1, 5, 5, 1, 5] 일때, 이를 정렬하면 [6, 5, 5, 5, 3, 1, 1, 0]이다.
i는 0부터 시작.
i가 3일 때, v == 5. i+1 == 4. 즉, v < i+1 성립되지 않음.
i가 4일 때, v == 3. i+1 == 5. 즉, v < i+1 성립.
i가 3일 때가 i+1편 이상의 논문에서 i+1번 이상 언급된 케이스 중 가장 i+1이 큰 경우이다.
만약 citations의 모든 수가 citations의 원소의 개수보다 큰 수일 경우에는 h의 최댓값은 citations의 원소의 개수가 된다.
결과
Best Code
def solution(citations):
citations.sort(reverse=True)
answer = max(map(min, enumerate(citations, start=1)))정말 짧지만 잘 짠 코드인 것 같다. 방법이 신기하다.
citations가 [3, 0, 6, 1, 5, 5, 1, 5]인 예시를 보이면 설명해보면, 내림차순 정렬했을 때 다음과 같다.
[6, 5, 5, 5, 3, 1, 1, 0]
이 리스트에서 enumerate(citations, start=1) 만 실행하게 된다면 다음과 같다.
[(1, 6), (2, 5), (3, 5), (4, 5), (5, 3), (6, 1), (7, 1), (8, 0)]
각각의 튜플에서 첫번째 원소는 순서, 두번째 원소는 인용된 횟수이다. 각 튜플별로 최솟값을 구해서 리스트를 만들면 다음과 같다.
[1, 2, 3, 4, 3, 1, 1, 0]
여기서 1~4는 첫번째 원소인 순서이고 그 이후부터는 인용된 횟수이다. 따라서 여기서 max를 써서 가장 큰 수를 가져오면 이것이 h값 중 가장 큰 값인 H-Index가 된다.
