[LeetCode] H-Index

If you've read the Wikipedia article of H-Index, there is already a neat formula there for computing the h-index, which is written below using the notations of the problem. Note that in the formula below, citations is sorted in descending order and i is 1-indexed.

h = max_i(min(i, citations[i]))

Now you will easily write down the following code.

class Solution {
public:
    int hIndex(vector<int>& citations) {
        sort(citations.rbegin(), citations.rend());
        int h = 0, i = 0;
        for (int c : citations)
            h = max(h, min(++i, c));
        return h;
    }
};

This code takes 20ms. In fact, rbegin and rend seems to be relatively slow. An alternative is to sort citations in normal ascending order and then count all those papers with citations larger than their indexes, as what Stefan does here. Now the code runs in 12ms.

class Solution {
public:
    int hIndex(vector<int>& citations) {
        sort(citations.begin(), citations.end()); 
        int h = 0, i = citations.size();
        for (int c : citations)
            h += (c > --i);
        return h;
    }
};

Well, both the above codes are in O(nlogn) time. Is there a linear time solution? The answer is yes: refer to this post if you like :-)

class Solution {
public:
    int hIndex(vector<int>& citations) {
        int n = citations.size(), h = 0;
        int* counts = new int[n + 1]();
        for (int c : citations)
            counts[min(c, n)]++;
        for (int i = n; i; i--) {
            h += counts[i];
            if (h >= i) return i;
        } 
        return h;
    }
};

This code uses both linear time and space, and runs in 8ms. Wow, we've moved a long way from the original 20ms version :-)