1. 问题描述:
给定一位研究者论文被引用次数的数组(被引用次数是非负整数)。编写一个方法,计算出研究者的 h 指数。h 指数的定义:h 代表“高引用次数”(high citations),一名科研人员的 h 指数是指他(她)的 (N 篇论文中)总共有 h 篇论文分别被引用了至少 h 次。且其余的 N - h 篇论文每篇被引用次数不超过 h 次。例如:某人的 h 指数是 20,这表示他已发表的论文中,每篇被引用了至少 20 次的论文总共有 20 篇。
示例:
输入:citations = [3,0,6,1,5]
输出:3
解释:给定数组表示研究者总共有 5 篇论文,每篇论文相应的被引用了 3, 0, 6, 1, 5 次。由于研究者有 3 篇论文每篇至少被引用了 3 次,其余两篇论文每篇被引用不多于 3 次,所以她的 h 指数是 3。
提示:如果 h 有多种可能的值,h 指数是其中最大的那个。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/h-index
2. 思路分析:
① 分析题目可以知道我们需要在一个范围([1, len(citations)])中找到一个最大的h指数mid,使得论文的被引用次数超过mid的论文数目n是大于等于mid的,很明显可以使用二分查找的方法解决,左边界为1,右边界为n,二分查找可以使用一个check方法来检查当前的mid是否满足题目的条件如果满足那么更新下一次二分查找的其中一个边界,否则更新另外一个边界(可以先排序一下这样可以在check方法中判断不满足的时候可以直接返回),这里为了避免边界的问题,我们可以声明一个res变量来记录h指数,当使用check函数返回的结果是满足题目条件的那么更新答案res即可,循环结束res记录的就是我们要求解的h指数。
② 除了①中的方法之外,其实我们可以先对citations由大到小排序,然后从后往前遍历只要是当前的citations[i] >= i + 1说明位置i前面的值都是大于等于i + 1的那么直接返回当前的i + 1即可(是否存在h个数大于等于h)。
3. 代码如下:
二分查找:
from typing import List
class Solution:
# 判断当前的h指数mid是否满足题目的条件
def check(self, mid, citations: List[int]):
n = 0
for i in range(len(citations) - 1, -1, -1):
if citations[i] >= mid: n += 1
else: return n >= mid
return n >= mid
def hIndex(self, citations: List[int]) -> int:
n = len(citations)
l, r = 0, n
res = 1
# 先排序这样在check函数判断不满足的时候可以直接返回
citations.sort()
while l <= r:
mid = (l + r) // 2
if self.check(mid, citations):
res = mid
l = mid + 1
else:
r = mid - 1
return res
from typing import List
class Solution:
def hIndex(self, citations: List[int]) -> int:
n = len(citations)
l, r = 0, n - 1
res = 0
while l <= r:
mid = (l + r) // 2
# n - mid - 1表示坐标的转换, 当满足条件之后说明答案肯定是在mid的右边
if citations[n - mid - 1] >= mid + 1:
l = mid + 1
res = mid + 1
else:
r = mid - 1
return res
from typing import List
class Solution:
def hIndex(self, citations: List[int]) -> int:
citations.sort(reverse=True)
for h in range(len(citations) - 1, -1, -1):
if citations[h] >= h + 1: return h + 1
return 0