最小的k个数(第k小的数)

1. 利用快排的思想,由于每次只选择左右部分中的一部分,因此时间复杂度为O(n).

# -*- coding:utf-8 -*-
class Solution:
    # (O(n)复杂度)
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        length = len(tinput)
        if k == length:
            return sorted(tinput)
        if k <= 0 or k > length:
            return []
        start = 0
        end = len(tinput) - 1
        return self.quick_sort(tinput, start, end, k)
    def quick_sort(self, ls, start, end, k):
        if start <= end:
            index1 = self.sort1(ls, start, end)
            if index1 == k:
                return sorted(ls[:k])
            elif index1 < k:
                return self.quick_sort(ls, index1+1, end, k)
            else:
                return self.quick_sort(ls, start, index1-1, k)
    def sort1(self, ls, start, end):
        i = start
        j = end
        base = ls[i]
        while i < j:
            while ls[j] > base and i < j:
                j -= 1
            if i < j:
                ls[i] = ls[j]
                i += 1
            while ls[i] < base and i < j:
                i += 1
            if i < j:
                ls[j] = ls[i]
                j -= 1
        ls[i] = base
        return i

2. 堆排序,当元素个数小于k时,直接填入直到元素个数等于k时,对已有的k个元素进行堆排序形成最大堆(Ologk),接下来当元素个数大于k时,每次都将新元素与堆顶比较,若比堆顶还大,则不可能为最小的k个数,pass;若比堆顶元素小,则替换堆顶元素,并重新调整堆的顺序(Ologk).总时间复杂度为O(nlogk). 由于堆元素个数固定,适用于数组很大的情况,可减少内存占用。

# -*- coding:utf-8 -*-
class Solution:
    # (O(n)复杂度)
    def GetLeastNumbers_Solution(self, tinput, k):
        length = len(tinput)
        ls = []
        if k == length:
            return sorted(tinput)
        if k > length or k < 1 or length <= 0:
            return []
        for i in range(length):
            
            if i < k:
                ls.append(tinput[i])
                if i == k-1: # 此时正好k个元素,开始初始化最大堆
                    start = (0+i)//2
                    for j in range(start, -1, -1):
                        print(j)
                        self.heap_once(ls, j, k-1)
                        print(ls)
            else:
                if tinput[i] < ls[0]:
                    ls[0] = tinput[i]
                    self.heap_once(ls, 0, k-1)
                else:
                    pass
        return sorted(ls)
    def heap_once(self, ls, start, end):
        i = start
        save = ls[start]
        j = i*2+1
        while j <= end:
            print(j)
            if j+1 <= end and ls[j+1] > ls[j]:
                j += 1
            if ls[i] < ls[j]:
                ls[i], ls[j]= ls[j], ls[i]
                i = j 
                j = 2*j+1
            else:
                break
        

 

在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、付费专栏及课程。

余额充值