最小的k个数

最小的k个数,某搜索公司的实习招聘中遇见这个题目,今天舍友讨论某电子商务公司校招也出了这个题目,说明是比较经典的数据结构问题,自己再回顾发现有点模糊,有必要再整理一遍。参考资料:《剑指offer》第五章,P167

题目

输入n个整数,找到最小的k个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4

解法1:堆排序,复杂度为O(n*log(k))

建一个大小为k的最大堆
对每一个数进行如下处理:

  • 如果大于或等于最大堆得根节点节点则直接pass
  • 否则如果小于最大根则将最大堆根节点替换为该数,调整堆,具体调整策略是父节点和两个子节点比较,将父节点和较大的子节点交换,递归下去,直至叶子节点

每一次堆调整的复杂度为log(k),每来一个数调整一次堆,所以复杂度为O(n*log(k)),可作为外排序,内存中保存堆就行了

解法2:如果可以修改输入数组,分区partition

快速排序算法中的分区(partition),找一个参考元素将数组分区,小于参考元素的放左边,大于参考元素的放右边,返回参考元素所在下标。如果返回下标小于k-1则,第k个最小元素在右边部分,对右边数组分区;否则如果下标大于k-1,第k个最小元素在左边部分,对左边数组进行分区;直至得到的分区下标为k-1,则数组中下标为[0,k-1]的元素为最小的k个元素
复杂度为o(n),因为需要对整个数组进行修改,所以只适用于内排序

在C语言中,找到一组整数中的最小k个数可以采用多种算法实现,其中一种常见的方法是使用优先队列(通常称为堆),特别是大顶堆(Max Heap)。这里提供一个简单的示例,使用大顶堆结构: ```c #include <stdio.h> #include <stdlib.h> // 定义一个数组大小 #define MAX_SIZE 100 // 结构体表示堆节点,包含值和索引 typedef struct { int value; int index; } MinHeapNode; // 大顶堆实现,用于存储前k小的元素 void max_heapify(int arr[], int n, int i) { int largest = i; // 初始化最大值位置为根节点 int left = 2 * i + 1; // 左孩子 int right = 2 * i + 2; // 右孩子 if (left < n && arr[left] > arr[largest]) { largest = left; } if (right < n && arr[right] > arr[largest]) { largest = right; } if (largest != i) { // 如果有更大值 swap(&arr[i], &arr[largest]); // 交换 max_heapify(arr, n, largest); // 递归调整子树 } } // 建立大顶堆 void build_max_heap(int arr[], int k) { for (int i = k / 2 - 1; i >= 0; i--) { max_heapify(arr, k, i); } } // 添加新元素到堆并保持堆性质 void insert(int arr[], int n, int k, int new_val, int new_index) { arr[n++] = new_val; // 添加新元素 max_heapify(arr, k, n - 1); // 调整以保持堆 } // 获取最小k个数 void get_min_k(int arr[], int k) { printf("The smallest %d numbers are:\n", k); for (int i = 0; i < k; i++) { printf("%d ", arr[0]); swap(&arr[0], &arr[k - 1]); // 将当前堆顶移到末尾 max_heapify(arr, k - 1, 0); // 更新堆 } } // 主函数示例 int main() { int arr[] = {9, 8, 7, 6, 5, 4, 3, 2, 1}; int n = sizeof(arr) / sizeof(arr[0]), k = 3; build_max_heap(arr, k); // 创建初始堆 // 假设我们有新元素插入 insert(arr, n, k, 100, 10); // 新元素:100, 索引:10 get_min_k(arr, k); // 输出前k小数 return 0; } ``` 在这个例子中,`build_max_heap()`函数建立了一个大顶堆,`insert()`函数用于添加新元素并维护堆属性,`get_min_k()`函数则从堆中获取并删除最小的k个元素。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值