数组中第K大的元素

寻找第K大

http://www.nowcoder.com/questionTerminal/e016ad9b7f0b45048c58a9f27ba618bf

利用快排中的Partition
算法导论中经典的partition

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        return findK(a, 0, n-1, K);
    }
    int findK(vector<int> &a, int L, int R, int K){
        int mid = partition(a, L, R);
        int nRight = R - mid + 1;
        if(K == nRight){
            return a[mid];
        }else if(K > nRight){
            return findK(a, L, mid - 1, K - nRight);
        } else{
            return findK(a, mid + 1, R, K);
        }
    }
    int partition(vector<int> &a, int L, int R){
        int v = a[R];
        int j = L-1;
        for(int i = L; i <= R; i++){
            if(a[i] <= v){
                j++;
                swap(a[i], a[j]);
            }
        }
        return j;
    }
};
全部评论

相关推荐

屌丝逆袭咸鱼计划:心态摆好,man,晚点找早点找到最后都是为了提升自己好进正职,努力提升自己才是最关键的😤难道说现在找不到找的太晚了就炸了可以鸡鸡了吗😤早实习晚实习不都是为了以后多积累,大四学长有的秋招进的也不妨碍有的春招进,人生就这样
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务