n个数里面找出最大的m个数字(快排思想)
分析:这个问题,我之前遇到的时候想到的解决方案是,最小堆解决方法。建立个数为m的最小堆,然后遍历n维护这个最小堆就可以了,算法的时间复杂度是n*log(m)。还是比较高效的算法的。
今天我又发现了一种解决方法,那就是STL里面的一种算法,STL里面的nth_element就是这样的一种算法。利用类似于快速排序的过程,找到前面的m个最大的数字。
不多说了,看代码吧。通过代码可以看出,和快排基本相似。
#include <iostream> #include <cstdio> using namespace std; //在m个数中寻找最大的n个数 //将数组M[m]中的最大的n(n<m)个数放在数组前面 void search(int M[], int m, int n) { if (n >= m) return; int t = M[0]; int i = 0, j = m - 1; while (i < j) { while (i < j && M[j] < t) --j; if (i < j) M[i++] = M[j]; while (i < j && M[i] >= t) ++i; if (i < j) M[j--] = M[i]; } M[i] = t; if (i < n) { search(&M[i+1], m - i - 1, n - i - 1); } else if (i > n) { search(M, i, n); } return; } void test() { int M[] = {8, 3, 9, 0, 4, 2, 5, 7, 1, 6}; int m = 10, n = 6; printf("原始数据:"); for(int i = 0; i < m; ++i) printf("%d ", M[i]); printf(" "); search(M, m, n); printf("结果数据:"); for(int i = 0; i < n; ++i) printf("%d ", M[i]); printf(" "); } int main() { test(); return 0; }
声明:该文观点仅代表作者本人,入门客AI创业平台信息发布平台仅提供信息存储空间服务,如有疑问请联系rumenke@qq.com。
- 上一篇:没有了
- 下一篇: C语言二进制文件读写以及大小端转换