之前去面试遇到了这个问题,题目:找出无序数组中第二大的数字。
定睛一看,好简单,一次遍历就可以找出第二大的数字,不过,这样写没有什么特别之处,因为实在是太简单了。自己仔细想想起了之前做过的题,看看有没有什么类似的。于是,想起了之前在网上看到的一道面试题:找出无序数组中最小的k个数。 要找出最小的k个数,可以用快速选择算法,只要在快速排序之后,枢纽刚好是第k+1个数,那么,他左边的数,刚好就是最小的k个数。要找出无序数组中第二大的数字,可以这样转换:
0)假设数组中有n个元素,a1~an,而我们要求第k大的数字,就相当于求排序后的第m = n - k + 1个数字。
1)假设一次快速排序之后,如果枢纽的是第x个(1~n)。
2)如果x = m,则这个枢纽ax就是我们要求的第k大的数字,结束;如果x < m,则对a0 ~ ax-1进行一次快速排序,重复第2)步;如果x > m,则对ax+1 ~ an 进行一次快速排序,重复第2)步。
快速选择的代码如下:
//a[] - 要排序的数组 //b - 要排序的子区间的开始索引 //e - 要排序的子区间的结束索引 //k - 求出排序后第k个数 int quickselect(int a[], int b, int e, int k){ int i = b ; int j = e + 1; int x = a[i]; while(true){ while(a[++i] < x && i < j); while(a[--j] > x); if(i >= j) break; else swap(a[i], a[j]); } a[b] = a[j]; a[j] = x; if(k - 1 == j) return a[j]; else if(k - 1 > j) return quickselect(a, j + 1, e, k); else return quickselect(a, b, j - 1, k); }
总结:快速选择可以用于这类面试题:
1)求出无序数组中第k大的数
2)求出无序数组中最大/小的k个数
版权声明:本文为博主原创文章,未经博主允许不得转载。