首页 > 代码库 > nth_element学习
nth_element学习
今天学习到STL中的nth_element，她是一个默认能求第k小的数的方法，需要的头文件为algorithm。
默认为：nth_element(start, start+n, end)
使第n大元素处于第n位置（从0开始,其位置是下标为n的元素），并且比这个元素小的元素都排在这个元素之前，比这个元素大的元素都排在这个元素之后，但不能保证他们是有序的。
1 #include <stdio.h> 2 #include <algorithm> 3 #include <iostream> 4 using namespace std; 5 6 int main() 7 { 8 int n,k; 9 int a[10];10 while(scanf("%d %d",&n, &k) == 2 )11 {12 for(int i=0; i<n; i++)13 scanf("%d",&a[i]);14 nth_element(a,a+k-1,a+n);15 printf("%d\n",a[k-1]);16 for(int i=0; i<n; i++)17 printf("%d ",a[i]);18 printf("\n");19 }20 return 0;21 }
当然，你还可以求第K大的数，方法即是添加一个greater<int>()参数，
1 #include <stdio.h> 2 #include <algorithm> 3 #include <iostream>//这里需要添加iostream 4 using namespace std; 5 6 int main() 7 { 8 int n,k; 9 int a[10];10 while(scanf("%d %d",&n, &k) == 2 )11 {12 for(int i=0; i<n; i++)13 scanf("%d",&a[i]);14 nth_element(a,a+k-1,a+n,greater<int>());15 printf("%d\n",a[k-1]);16 for(int i=0; i<n; i++)17 printf("%d ",a[i]);18 printf("\n");19 }20 return 0;21 }
如上图所示，我要求第2大的数，首先输出了4，这是对的！
然后，nth_element将4放在了第2个位置，比起大的，均在其前；比其小的，均在其后。但并不一定有序！
这也就造成了nth_element的O（n）的逆天复杂度。
nth_element学习
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。