[코테] 99클럽 코테 스터디 8일차 TIL - 정렬

Question1 - [미들러] H 인덱스

 

문제 설명 https://school.programmers.co.kr/learn/courses/30/lessons/42747

 

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

 

---

문제풀이

 

trial1 : 반복문 (ON^2)

 

가장 먼저 푼 방법은 정말 정직하게 반복문을 두번 돌린 케이스로, citation의 길이+1만큼 돌아가면서 h로 성립되는 경우들을 리스트에 모두 담고, 그 리스트의 max를 파악하는 방식이다.

def solution(citations):
    lst = []
    for h in range(len(citations)+1):
        count = 0
        for i in citations : 
            if i >= h : 
                count += 1 
        if count >= h : 
            lst.append(h)
    return max(lst)

 

 

trial2 : 순서가 있는 정렬 O(nlogn)

 

풀이 속도를 더 빠르게 만들기 위해서는, 반대로 우선 citations를 내림차순으로 높은 것부터 정렬한 뒤에, 

citation의 횟수가 그 순서(enumerate 사용) 이상인 경우 h 인덱스는 그 값이 된다.

def solution(citations):
    citations.sort(reverse=True) # 높은것부터 
    h_index = 0

    for i, citation in enumerate(citations) : 
        if citation >=i+1 : 
            h_index = i+1
        else : 
            break 

    return h_index

 

 

trial3 : 이진트리 

2와 비슷한 방법으로, 이진트리로 바꿔서 풀수도 있다. 실행 결과 또한 동일하다.

import bisect

def solution(citations):
    sorted_citations = []
    h_index = 0

    for citation in citations:
        bisect.insort(sorted_citations, citation)

        # H-지수를 계산하기 위해 현재 리스트의 길이와 각 요소를 확인
        current_h = 0
        for i in range(len(sorted_citations)):
            if sorted_citations[i] >= len(sorted_citations) - i:
                current_h = len(sorted_citations) - i
                break
        h_index = max(h_index, current_h)
    
    return h_index

 

---

 

Question1 - [챌린저] orderly queue

 

(https://leetcode.com/problems/orderly-queue/description/)