動機

把二分搜最討厭的部分完全展示的一題

Problem

Given an array of integers citations where citations[i] is the number of citations a researcher received for their ith paper and citations is sorted in an ascending order, return compute the researcher's h-index.

According to the definition of h-index on Wikipedia: A scientist has an index h if h of their n papers have at least h citations each, and the other n − h papers have no more than h citations each.

If there are several possible values for h, the maximum one is taken as the h-index.

You must write an algorithm that runs in logarithmic time.

 

Example 1:

Input: citations = [0,1,3,5,6]Output: 3Explanation: [0,1,3,5,6] means the researcher has 5 papers in total and each of them had received 0, 1, 3, 5, 6 citations respectively.Since the researcher has 3 papers with at least 3 citations each and the remaining two with no more than 3 citations each, their h-index is 3.

Example 2:

Input: citations = [1,2,100]Output: 2

 

Constraints:

  • n == citations.length
  • 1 <= n <= 105
  • 0 <= citations[i] <= 1000
  • citations is sorted in ascending order.

Sol

h-index的定義是

前N個paper的引用數大於等於N

因為已經排序了,所以第i個,代表前i個的引用數都小於等於他

所以h-index就是總長減i

但痛苦的是

  1. i是index
  2. h-index的range是 0~總長,故i的範圍是0~總長

如果要處理的漂亮要在

  • 終止條件
  • 算中點
  • 縮小範圍時

下功夫

Ver1

  • 終止條件: 包含右邊
  • 算中點: 無條件進位
  • 縮小範圍時: 不保留原本的index,兩邊都往內縮
class Solution:
    def hIndex(self, cs: List[int]) -> int:
        if not cs:
            return 0
        i = 0
        j = len(cs)-1
        while i <= j:
            mid = (i+j)//2 if (i+j) % 2 == 0 else (i+j+1)//2 # 無條件進位
            print(i,j,mid)
            if cs[mid] == len(cs) - mid:
                return cs[mid]
            elif cs[mid] > len(cs) - mid:
                j = mid-1
            else:
                i = mid+1
        print(i,j)
        return len(cs)-i

Ver2

  • 終止條件: 不包含右邊
  • 算中點: 無條件捨去
  • 縮小範圍時: 保留右邊的index
class Solution:
    def hIndex(self, cs: List[int]) -> int:
        if not cs:
            return 0
        i = 0
        j = len(cs)
        while i < j:
            mid = (i+j)//2 # 無條件捨去
            if cs[mid] == len(cs) - mid:
                return cs[mid]
            elif cs[mid] > len(cs) - mid:
                j = mid
            else:
                i = mid+1
        return len(cs)-i

Ref

[Algorithm, Leetcode] H-Index II 求H指数之二