본문 바로가기
CodingTest

[코딩테스트 JAVA]H-Index 정렬 - Arrays.stream(..).filter(..).count()

by KkingKkang 2025. 2. 3.
728x90

문제 설명
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입니다.


푸는것 보다 문제 해석이 더 어려웠음. .

처음에 논문의 인용 횟수만 배열로 주어진다면, 논문의 번호나 고유 넘버는 어디있지? 라고 생각했다. 

결론적으로, 논문 h를 1로 놓고 1부터 시작해서 조건이 만족되는 한 계속 증가하다가, 처음으로 조건이 깨지는 순간 멈추면 되는 것이었다.

h편 이상의 논문이 h번 이상 인용된 지점을 찾으면 됨.

예를 들어, [10, 8, 5, 4, 3] 일 경우

h 인용 횟수 >= h인 논문 개수 가능한 요소 o/x
1 5 [10, 8, 5, 4, 3]  o
2 5 [10, 8, 5, 4, 3]  o
3 5 [10, 8, 5, 4, 3]  o
4 4 [10, 8, 5, 4]  o
5 3 [10, 8, 5]  x

h가 될 수 있는 최대값은 4이다.


import java.util.Arrays;

class Solution {

	public static int countGreater(int[] arr, int n) {
    	return (int)Arrays.stream(arr).filter(num -> num >= n).count();
    }
    
    public int solution(int[] citations){
    	int answer = 0;
        int hIndex = 0;
        
        Arrays.sort(citations); 
        
        while(true) {
        
        	if(answer < hIndex) break;
            answer = countGreater(citations, hIndex);
            hIndex++;
       }
       
       return answer;
   }
}

(int)Arrays.stream(arr).filter(num -> num >= n).count();

가 사기인거 같다. 이에 대해 자세히 알아보자.

1. Arrays.stream(arr)

- 배열을 스트림으로 변혼 : Java Stream API를 통해 스트림으로 변환

- 스트림이란 ? : 데이터의 흐름을 나타내며, 컬렉션(배열, 리스트 등)의 요소를 처리할 수 있는 일련의 연산을 가능하게 함

- 결과 : IntStream 이라는 정수 스트림이 생성됨 

 

2. .filter(num -> num >=n)

- 필터링 함수 : filter 메서드는 스트림의 각 요소를 조건에 따라 걸러내는 역할을 함

- 람다 표현식 : num -> num >= n 은 람다 함수로, num은 배열의 각 요소를 의미한다. 

- 결과 : n보다 크거나 같은 숫자들만 포함된 새로운 스트림이 생성됨 

 

3. .count() 

- count() 메서드는 스트림에 남아 있는 요소의 개수를 long 타입으로 반환함

- long타입으로 반환하기 때문에 int 타입으로 변환해야 함. 

728x90

댓글