牛客网编程高频题8——NC88寻找第K大

目录

寻找第K大

题目描述

示例1

输入

返回值

方法：快速排序

方法二：堆排序

方法三：归并排序

---

 

 

寻找第K大

题目描述

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。

给定一个整数数组a，同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

示例1

输入

[1,3,5,2,2],5,3

返回值

2

 

方法：快速排序

思路就是快速排序的划分，每次划分都会最终确定一个元素的位置，

• 如果这个位置正好是第K大数的位置，那么就找到了
• 如果小于第K大数的位置，则在右边继续寻找
• 否则，在左边继续寻找

import java.util.*;

public class Solution {
    public int findKth(int[] a, int n, int K) {
        // write code here
        int res=QuickSort(a,0,n-1,K);
        return res;
    }
    
    public static int QuickSort(int[] arr,int low,int high,int K){
        if(low<high){
            int partition=Partition(arr,low,high);
            if(K-1==partition) {
                return arr[partition];
            }else if(K-1<partition){
                QuickSort(arr,low,partition-1,K);
            }else {
                QuickSort(arr, partition + 1, high, K);
            }
        }
        return arr[K-1];
    }

    public static int Partition(int[] arr,int low,int high){//从大到小排列
        int curVal=arr[low];
        int curPos=low;
        for (int i = low+1; i <= high; i++) {
            if(arr[i]>curVal){
                curPos++;
                if(curPos!=i){
                    swap(arr,curPos,i);
                }
            }
        }
        arr[low]=arr[curPos];
        arr[curPos]=curVal;
        return curPos;
    }

    public static void swap(int[] arr,int i,int j){
        int temp=arr[i];
        arr[i]=arr[j];
        arr[j]=temp;
    }
}

快排的时间和空间复杂度

 

方法二：堆排序

为了巩固之前学习的知识，写了一遍堆排序

import java.util.*;

public class Solution {
    public int findKth(int[] a, int n, int K) {
        // write code here
        int res=heapSort(a,K);
        return res;
    }
    
    //堆排序
    public static int heapSort(int[] arr,int K){
        for (int i = arr.length/2-1; i >=0 ; i--) {//从第一个非叶子节点开始调整
            adjustHeap(arr,i,arr.length);
        }

        for (int j = 1; j < K; j++) {
            swap(arr,0,arr.length-j);
            adjustHeap(arr,0,arr.length-j);
        }
        return arr[0];
    }

    public static void adjustHeap(int[] arr,int i,int length){
        int temp=arr[i];
        for (int j = 2*i+1; j < length; j++) {
            if(j+1<length&&arr[j]<arr[j+1]){
                j++;
            }
            if(arr[j]>temp){
                swap(arr,i,j);
                i=j;
            }
        }
    }

    public static void swap(int[] arr,int i,int j){
        int temp=arr[i];
        arr[i]=arr[j];
        arr[j]=temp;
    }
}

 

方法三：归并排序

巩固排序算法知识

import java.util.*;

public class Solution {
    public int findKth(int[] a, int n, int K) {
        // write code here
        mergeSort(a,0,a.length-1);
        return a[K-1];
    }
    
    //归并排序
    public static void mergeSort(int[] arr,int low,int high){
        if(low<high){
            int mid=(low+high)/2;
            mergeSort(arr,low,mid);
            mergeSort(arr,mid+1,high);
            merge(arr,low,mid,high);
        }
    }

    public static void merge(int[] arr,int low,int mid,int high){
        int[] temp=new int[arr.length];
        for (int i = 0; i < arr.length; i++) {
            temp[i]=arr[i];
        }

        int i=low;
        int j=mid+1;
        int k=low;
        while(i<=mid&&j<=high){
            if(temp[i]>temp[j]){
                arr[k++]=temp[i++];
            }else{
                arr[k++]=temp[j++];
            }
        }
        while(i<=mid){
            arr[k++]=temp[i++];
        }
        while(j<=high){
            arr[k++]=temp[j++];
        }
    }
}