C语言实现求第N大的元素(第N小的元素同理)
思路(以全局最大值举例):
找到全局最小值
找到全局最大值
将全局最大值赋值为 全局最小值-1
重复2-3步骤N-1次(N为第N大的N)
再次找到最大值,此最大值即为 第N大元素
目的:
这样写的好处就是时间复杂度低,对于大量的数据,比起先排序再取第N最大值来说效率要高得多
优化:
针对不同的要求以及不同的环境,可以进行不同的优化。
如:数据都为正数,则可直接将最大值赋值为小于0的任意的你喜欢♥的值。
以下为示例代码:
比较函数,返回最大值或者最小值的下标:
int compare(int * arr, int n, int flag) {
int m = arr[0], mi = 0;
for (size_t i = 0; i < n; i++)/* true为max, false为min */
if (flag ? (m < arr[i]) : (m > arr[i])) m = arr[i], mi = i;
return mi;
}
核心赋值代码:
int min = arr[compare(arr, ARRN, 0)] - 1; /*找到全局最小值,并令其-1*/
for (size_t i = 0; i < maxN - 1; i++) { /* maxN为第N大的值 */
arr[compare(arr, ARRN, 1)] = min;
}
总代码:
#include <stdio.h>
const int ARRN = 10;
int compare(int * arr, int n, int flag);
int main() {
int maxN = 3; /*第N大*/
int arr[ARRN] = { 5,2,0,1,3,1,4,6,7,8 };
int min = arr[compare(arr, ARRN, 0)] - 1; /*找到全局最小值,并令其-1*/
for (size_t i = 0; i < maxN - 1; i++) { /* maxN为第N大的值 */
arr[compare(arr, ARRN, 1)] = min;
}
printf("第%d大的数是%d\n", maxN, arr[compare(arr, ARRN, 1)]);
return 0;
}
int compare(int * arr, int n, int flag) {
int m = arr[0], mi = 0;
for (size_t i = 0; i < n; i++)/* true为max, false为min */
if (flag ? (m < arr[i]) : (m > arr[i])) m = arr[i], mi = i;
return mi;
}