二分查找题目：H 指数 II

题目

标题和出处

标题：H 指数 II

出处：275. H 指数 II

难度

5 级

题目描述

要求

给定一个整数数组 $\text{citations}$，其中 $\text{citations[i]}$ 表示研究者的第 $\text{i}$ 篇论文被引用的次数，$\text{citations}$ 按照升序排序，计算并返回该研究者的 h 指数。

根据 h 指数的定义，一名科研人员的 h 指数是指其发表的 $\text{n}$ 篇论文中总共有 $\text{h}$ 篇论文分别被引用了至少 $\text{h}$ 次，且其余的 $\text{n}-\text{h}$ 篇论文每篇被引用次数不超过 $\text{h}$ 次。

如果 $\text{h}$ 有多种可能的值，其中最大的值作为 h 指数。

请使用对数时间复杂度的算法解决此问题。

示例

示例 1：

输入：$\text{citations = [0,1,3,5,6]}$
输出：$\text{3}$
解释：$\text{[0,1,3,5,6]}$ 表示研究者总共有 $\text{5}$ 篇论文，每篇论文相应的被引用了 $\text{0, 1, 3, 5, 6}$ 次。
由于研究者有 $\text{3}$ 篇论文每篇至少被引用了 $\text{3}$ 次，其余 $\text{2}$ 篇论文每篇被引用不多于 $\text{3}$ 次，所以 h 指数是 $\text{3}$。

示例 2：

输入：$\text{citations = [1,2,100]}$
输出：$\text{2}$

数据范围

• $\text{n}=\text{citations.length}$
• $\text{1}\le \text{n}\le {\text{10}}^{\text{5}}$
• $\text{0}\le \text{citations[i]}\le \text{1000}$
• $\text{citations}$ 按照升序排序

解法

思路和算法

这道题是「H 指数」的延伸，数组 $\text{citations}$ 已经按照升序排序。利用数组有序的特点，可以使用二分查找计算 h 指数，时间复杂度是 $O(\log n)$，其中 $n$ 是论文总数。

根据 h 指数的定义可知，在升序数组 $\text{citations}$ 中，对于 $0\le x\le n$，被引用次数不少于 $\text{citations}[x]$ 的论文数量为 $n-x$，这里假设 $\text{citations}[n]=+\infty$。寻找符合 $\text{citations}[x]\ge n-x$ 的最小的 $x$，则 $n-x$ 即为 h 指数。

由于 $\text{citations}[x]$ 和 $n-x$ 都有关于 $x$ 的单调性，$\text{citations}[x]$ 关于 $x$ 单调递增，$n-x$ 关于 $x$ 单调递减，因此可以使用二分查找计算最大的 $x$ 的值。

用 $\text{low}$ 和 $\text{high}$ 分别表示二分查找的下标范围的下界和上界，初始时 $\text{low}=0$，$\text{high}=n$。

每次查找时，取 $\text{mid}$ 为 $\text{low}$ 和 $\text{high}$ 的平均数向下取整，执行如下操作。

• 如果 $\text{citations}[\text{mid}]\ge n-\text{mid}$，则最小的 $x$ 小于等于 $\text{mid}$，因此在 $[\text{low},\text{mid}]$ 中继续寻找。

• 如果 $\text{citations}[\text{mid}]<n-\text{mid}$，则最小的 $x$ 大于 $\text{mid}$，因此在 $[\text{mid}+1,\text{high}]$ 中继续寻找。

当 $\text{low}=\text{high}$ 时，查找结束，此时 $\text{low}$ 即为最小的 $x$，h 指数为 $n-\text{low}$。

代码

class Solution {
    public int hIndex(int[] citations) {
        int n = citations.length;
        int low = 0, high = n;
        while (low < high) {
            int mid = low + (high - low) / 2;
            if (citations[mid] >= n - mid) {
                high = mid;
            } else {
                low = mid + 1;
            }
        }
        return n - low;
    }
}

复杂度分析

• 时间复杂度：$O(\log n)$，其中 $n$ 是数组 $\text{citations}$ 的长度。二分查找的时间复杂度是 $O(\log n)$。

• 空间复杂度：$O(1)$。