coding test

[파이썬, Java] H-Index

잔망루피 2020. 12. 1. 16:34

문제 설명

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일 테스트 케이스가 추가되었습니다.


  1. https://en.wikipedia.org/wiki/H-index 위키백과 

 

 

♡ 나의 풀이

 

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