leetcode Third Maximum Number

如何求无序数组中最大的k个数系列1

这道题是求数组中第三大的数。

方法一：C++的set容器的特点是内部自动有序且不包含重复元素。本题可以保持一个大小为3的set容器。每次向set中添加元素，添加完后若元素个数大于3，则从set中删除最小的元素。保留的永远是数组当前最大的3个元素。

AC代码 

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        if(nums.size()==0){
            return 0;
        }
        set<int> s;
        for(auto num:nums){
            s.insert(num);      // 必须先加入，再判断，否则先判断并从集合中删除一个元素，但是后来的元素又是重复就会导致集合中元素个数不足三个
            if(s.size()>3){
                s.erase(s.begin());
            }
        }
        if(s.size()<3){
            return *s.rbegin();
        }else{
            return *s.begin();   // 不能写成*s.end()，因为s.end()为空指针
        }
    }
};

另外，这里的维持一个仅仅保留3个元素的集合的思想很重要。在以下提到的非重复元素数组选取第k大的数的最大堆解法中，同样用到了这种思想。

 

对于非重复元素数组的选取第k大的数还有另外几种解法

方法2：分治法。

利用快排里面的partition函数，选择一个主元，将一个无序数组分成两个部分，前面的部分小于主元，后面部分大于主元。

    int partition(vector<int>& nums){
        int key = nums[nums.size()];
        int i=-1, j;
        for(j=0;j<nums.size()-1;j++){
            if(nums[j]<=key){
                i++;
                int temp = nums[j];
                nums[j] = nums[i];
                nums[i] = temp;
            }
        }
        nums[i+1] = key;
        return i+1;
    }

方法2：随机法。

这个算法同样来自算法导论，思想同样和快速排序类似。但这个算法中涉及到随机数生成器，因此它是一个随机算法。

算法导论上的伪代码是为了选择数组中第k小的元素。

伪代码如下：

randomized-select(A,p, r,i)
/*
p为数组的起始位置，r为数组结束位置
i即要选择的第i小的元素
*/
    
    if p == r
        return A[p]
    q = randomized-partition(A, p, r)
    k = q - p + 1        // 小于等于主元的元素个数
    if i == k
        return A[q]
    else if i < k
        return randomized-select(A, p, q, i)
    else 
        return randomized-select(A, q+1, r, i - k) 
    

暂时先写这么多，后面遇到具体例子，再把代码补全。