第K小元素

算法分类:

分治


算法原理:

每次丢弃一部分数组中元素,则问题规模以几何级数递减。

算法步骤:
1.对数组进行分组,分成小于data[k]元素,data[k],以及大于data[k]元素3组。
2.判断小于data[k]的元素个数num。
(1) 若num+1 == k,则返回data[k](获得答案)
(2) 若num + 1 < k,则对大于data[k]组元素递归,同时更改相应k值。
(3) 若num + 1 > k,则对小于data[k]组递归。

时空复杂度:

时间复杂度O(n)

代码实现:

#include <iostream>
using namespace std;

int FindMinK(int data[], int l, int r, int k)
{
	int  mm, i, j, temp, num;
	// 获取目前第k小元素 
	mm = data[l+k-1];
	// 将该元素放到数组末尾 
	swap(data[r],data[l+k-1]);
	
	i = j = l;
	while (j < r)
	{
		if(data[j] < mm)
		{
			swap(data[i],data[j]);
			++ i;	
		}	
		++ j;
	}
	swap(data[i],data[j]);
	
	num = i-l+1;
	if (k == num)
		return mm;
	else if (k > num)
		return FindMinK(data, i+1, r, k-num);
	else
		return FindMinK(data, l, i-1, k);
	
};

int main()
{
    
    //for (int i = 1; i <= sizeof(data)/sizeof(data[0]); ++i)
    //    printf("%d\n", findKMin(data, 0, 8, i));
    for (int i = 1; i <= 8; ++ i)
	{
		int data[]={8,6,6,4,4,3,2,1};  
    	printf("%d\n", FindMinK(data, 0, 7, i));
	}
	getchar();
    return 0;
}


要输出第k小元素,可以使用数组排序或堆排序算法来实现。 方法1:数组排序 首先将给定的数组进行排序,可以使用常见的排序算法如快速排序、冒泡排序等。排序完成后,可以直接通过索引找到第k小的元素。 示例代码: ```c #include <stdio.h> void swap(int *a, int *b) { int temp = *a; *a = *b; *b = temp; } void bubbleSort(int arr[], int n) { for (int i = 0; i < n-1; i++) { for (int j = 0; j < n-i-1; j++) { if (arr[j] > arr[j+1]) { swap(&arr[j], &arr[j+1]); } } } } int main() { int arr[] = {5, 9, 3, 1, 7}; int k = 2; int n = sizeof(arr) / sizeof(arr[0]); bubbleSort(arr, n); printf("第%d小的元素是:%d\n", k, arr[k-1]); return 0; } ``` 方法2:堆排序 可以利用最小堆来实现。首先将数组构建成最小堆,然后重复k-1次从堆顶弹出元素的操作,最后堆顶元素就是第k小的元素。 示例代码: ```c #include <stdio.h> void swap(int *a, int *b) { int temp = *a; *a = *b; *b = temp; } void heapify(int arr[], int n, int i) { int smallest = i; int left = 2 * i + 1; int right = 2 * i + 2; if (left < n && arr[left] < arr[smallest]) { smallest = left; } if (right < n && arr[right] < arr[smallest]) { smallest = right; } if (smallest != i) { swap(&arr[i], &arr[smallest]); heapify(arr, n, smallest); } } int findKthSmallest(int arr[], int n, int k) { for (int i = n / 2 - 1; i >= 0; i--) { heapify(arr, n, i); } for (int i = 0; i < k - 1; i++) { swap(&arr[0], &arr[n-1]); n--; heapify(arr, n, 0); } return arr[0]; } int main() { int arr[] = {5, 9, 3, 1, 7}; int k = 2; int n = sizeof(arr) / sizeof(arr[0]); int result = findKthSmallest(arr, n, k); printf("第%d小的元素是:%d\n", k, result); return 0; } ``` 以上是两种输出第k小元素的方法和示例代码。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值