LeetCode-Python-275. H指数 II

给定一位研究者论文被引用次数的数组(被引用次数是非负整数),数组已经按照升序排列。编写一个方法,计算出研究者的 h 指数。

h 指数的定义: “h 代表“高引用次数”(high citations),一名科研人员的 h 指数是指他(她)的 (N 篇论文中)至多有 h 篇论文分别被引用了至少 h 次。(其余的 N - h 篇论文每篇被引用次数不多于 次。)"

示例:

输入: citations = [0,1,3,5,6]
输出: 3 
解释: 给定数组表示研究者总共有 5 篇论文,每篇论文相应的被引用了 0, 1, 3, 5, 6 次。
     由于研究者有 3 篇论文每篇至少被引用了 3 次,其余两篇论文每篇被引用不多于 3 次,所以她的 h 指数是 3

说明:

如果 有多有种可能的值 ,h 指数是其中最大的那个。

进阶:

  • 这是 H指数 的延伸题目,本题中的 citations 数组是保证有序的。
  • 你可以优化你的算法到对数时间复杂度吗?

思路:

二分查找。

根据题意:找的是某个点,满足这个点的值 > = 它和它右侧的点个数之和cnt情况下,cnt就是当前的h指数

如果已经有h指数了,就可以试着往左找,因为越左cnt越大,h指数也越大,所以左侧可能存在更大的h值,

如果目前还没有h指数,就只能试着缩小cnt,往右找,

此题缩小搜索范围的思路就可以用二分查找来实现。

class Solution(object):
    def hIndex(self, citations):
        """
        :type citations: List[int]
        :rtype: int
        """
        l = len(citations)
        lo, hi = 0, l - 1
        res = 0
        while(lo <= hi):
            mid = lo + (hi - lo) // 2
            cnt = l - mid #包括mid自身右边还有的元素个数
            if citations[mid] >= cnt: 
                res = cnt
                hi = mid -1
            else:
                lo = mid + 1
        return res                         
                

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/87982177