首页 > 代码库 > 输出数组第k大的元素
输出数组第k大的元素
用快速排序的思想输出数组第k大的元素:
1 #include<iostream> 2 #include<algorithm> 3 using namespace std; 4 5 //递归实现:返回数组第k大的值.数组下标区间是[begin,end].其中数组假定n个元素,则k的值在区间[1,n]. 6 //能够使用这种方法的前提条件是:n个数不能重复。如果n个数中有重复,那么区间的大小不能保证就是第K大。 7 int findkth(int* arr, int begin, int end, int k) 8 { 9 int key=arr[begin]; 10 int i=begin,j=end; 11 while(i<j) 12 { 13 while(i<j && arr[j]<=key) j--; 14 arr[i]=arr[j]; 15 while(i<j && arr[i]>=key) i++; 16 arr[j]=arr[i]; 17 } 18 arr[i]=key; 19 20 //递归的结束条件为某一记录刚好划分到第k个位置 21 //由于数组从0开始,所以需要加1了 22 if (i == k-1)return arr[i]; 23 else if (i < k-1) return findkth(arr, i+1, end, k); 24 else return findkth(arr, begin, i-1, k); 25 } 26 27 28 //只做了一轮划分,没有递归完成快速排序.返回的是划分的下标。数组a[]下标区间[low,high] 29 template <class T> 30 int partition(T a[],int low,int high) 31 { 32 T key=a[low]; 33 int i=low,j=high; 34 while(i<j) 35 { 36 while(i<j && a[j]>=key) j--; 37 a[i]=a[j]; 38 while(i<j && a[i]<=key) i++; 39 a[j]=a[i]; 40 } 41 a[i]=key; 42 return i; 43 } 44 45 //利用快速排序的思想查找第K大的数值 46 //第一个参数代表要查找的数组 47 //第二个参数代表数组的长度 48 //第三个参数代表要查找第几大的元素 49 //非递归 50 template <class T> 51 T findkth2(T a[],int n,int k) 52 { 53 int low=0,high=n-1; 54 while(1) 55 { 56 int pos=partition(a,low,high); 57 if(pos==n-k) return a[pos]; 58 else if(pos<n-k) 59 { 60 low=pos+1; 61 high=n-1; 62 } 63 else if(pos>n-k) 64 { 65 high=pos-1; 66 low=0; 67 } 68 } 69 } 70 71 int main() 72 { 73 int a[11]={78934,234,65,32,543,354,567,3412,23,547,423}; 74 int b[11]={78934,234,65,32,543,354,567,3412,23,547,423}; 75 int c[11]={78934,234,65,32,543,354,567,3412,23,547,423}; 76 77 for(int i=0;i<11;i++) cout<<a[i]<<" "; 78 cout<<endl; 79 sort(a,a+11); 80 for(int i=0;i<11;i++) cout<<a[i]<<" "; 81 cout<<endl; 82 83 for(int i=11;i>=1;i--) 84 cout<<findkth(b,0,10,i)<<" "; 85 cout<<endl; 86 87 for(int i=11;i>=1;i--) 88 cout<<findkth2(c,11,i)<<" "; 89 cout<<endl; 90 91 92 cout<<endl<<endl; 93 for(int i=0;i<11;i++) cout<<b[i]<<" "; 94 cout<<endl; 95 for(int i=0;i<11;i++) cout<<c[i]<<" "; 96 cout<<endl; 97 return 0; 98 }
参考:http://blog.csdn.net/guangwen_lv/article/details/39674241
利用快速排序的特点:第一遍排序会确定一个数的位置,这个数左边都比它大,右边都比他小(降序),当左边区间大于K时,说明我们求的第K大数在左边区间,这时我们可以舍弃右边区间,将范围缩小到左边区间从而重复上述过程,直到确定一个数的位置时,左边区间的小是K-1那么这个数字就是我们所求。右边同理。
能够使用这种方法的前提条件是:n个数不能重复。如果n个数中有重复,那么区间的大小不能保证就是第K大。可以用HASH判重来将重复的数据去掉。然后再使用上述方法。
也可阅读如下另一位网友的java代码:
1 /** * Created with IntelliJ IDEA. 2 * User: hqtc 3 * Date: 16-3-22 4 * Time: 下午9:52 5 * To change this template use File | Settings | File Templates. 6 */ 7 public class QuickSort{ 8 9 private int getSeparatorIndex(int[] nums, int low, int high) { 10 int k = nums[low]; 11 while (low < high) { 12 while (k < nums[high] && low < high) { 13 high--; 14 } 15 if (low < high) 16 nums[low++] = nums[high]; 17 while (k > nums[low] && low < high) { 18 low++; 19 } 20 if (low < high) 21 nums[high--] = nums[low]; 22 } 23 nums[low] = k; 24 return low; 25 } 26 27 public void sort(int num[], int low, int high) { 28 if (low < high) { 29 int sepIndex = getSeparatorIndex(num, low, high); 30 sort(num, low, sepIndex - 1); 31 sort(num, sepIndex + 1, high); 32 } 33 } 34 35 public int findKthMax(int arr[], int k, int left, int right) { 36 int sepIndex = getSeparatorIndex(arr, left, right); 37 if (k == arr.length - sepIndex) { 38 return arr[sepIndex]; 39 } else { 40 if (k > arr.length - sepIndex) { 41 return findKthMax(arr, k, left, sepIndex - 1); 42 } else { 43 return findKthMax(arr, k, sepIndex + 1, right); 44 } 45 } 46 } 47 } 48 49 /*作者:环球探测 50 链接:http://www.jianshu.com/p/9f887dfc374a 51 來源:简书 52 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。*/
输出数组第k大的元素
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。