문제 설명
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입니다.
※ 공지 - 2019년 2월 28일 테스트 케이스가 추가되었습니다.
♡ 나의 풀이
def solution(citations):
length=len(citations)
citations.sort()
for i in citations:
cnt=length-citations.index(i)
if cnt <= i: # i번 이상 인용된 논문이 i편 이상
return cnt
return 0
length-citations.index(i)는 i번째 값(인용횟수 h)보다 인용 횟수가 큰 논문의 갯수(h번 이상 인용된 논문)다.
citations가 [0]이면 cnt=1이 아닌 0이 되어야하므로 return 0을 넣었다.
import java.util.*;
class Solution {
public int solution(int[] citations) {
int answer = 0;
int size=citations.length; // 논문의 수
Arrays.sort(citations);
for(int i=0; i<size; i++){
if(citations[i] >= size-i) return size-i;
}
return answer;
}
}
자바로도 구현
★ 다른 사람 풀이
def solution(citations):
length=len(citations)
citations.sort()
for i in range(length):
if length-i <= citations[i]:
return length-i
return 0
나와 같은 알고리즘을 썼지만 for문을 보면 i를 인덱스로 쓰고 있다.
index함수 안 써도 돼서 좀 더 효율적임.
def solution(citations):
citations.sort(reverse=True)
answer = max(map(min, enumerate(citations, start=1)))
return answer
리스트 citations를 내림차순으로 정렬했다. enumerate에 start를 1로 줘서 citations의 index가 1부터 시작이다.
min으로 index와 값 중 작은 값을 반환한다.
리스트 길이-인덱스 연산과 결과가 같다.
ex)
citations= [6, 5, 3, 1, 0]
index는 1 2 3 4 5
결과는 1, 2, 3, 1, 0이고 이 중 가장 큰 값인 3이 answer로 반환된다.
문제 출처 👉 프로그래머스
'coding test' 카테고리의 다른 글
[파이썬] 카펫 (0) | 2020.12.02 |
---|---|
[파이썬] 구명보트 (0) | 2020.12.02 |
[파이썬] 2016년 (0) | 2020.11.30 |
★ [파이썬] 문자열 압축 (0) | 2020.11.27 |
[파이썬, Java] 크레인 인형뽑기 게임 (0) | 2020.11.27 |