查找最小的k个元素

/**
查找最小的k个元素
题目:输入n个整数,输出其中最小的k个。
例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。
*/
/*非常传统的一道题*/
#include <stdio.h>
#include <stdlib.h>

#define N 9
#define K 4

void HeapAdjust(int a[],int s,int k)
{
    int t,key;

    key=a[s];
    for(t=2*s;t<=k;t*=2)
    {
        if(t<k && a[t]<a[t+1])
            t++;
        if(a[s]>a[t])
            break;
        a[s]=a[t];
        s=t;
    }
    a[s]=key;
}

void MinKNum(int a[],int n,int k)
{
    int i,tmp;

    if(n-1<k)
        exit(EXIT_FAILURE);
    
    for(i=k/2;i>0;i--)
        HeapAdjust(a,i,k);
    for(i=k/2+1;i<n;i++)
    {
        if(a[i]<a[1])
        {
            tmp=a[i];
            a[i]=a[1];
            a[1]=tmp;
        }
        HeapAdjust(a,1,k);
    }
}
int main()
{
    int a[N];
    int i;

    for(i=1;i<N;i++)//数组从a[1]开始存储
        scanf("%d",&a[i]);

    MinKNum(a,N,K);//建立大顶堆

    for(i=1;i<=K;i++)
        printf("%d ",a[i]);
}

您好,关于Python实现查找最小的k个数,我可以提供以下两种常见的方法: 方法一:使用堆 将数组前k个元素建立小根堆,然后遍历剩余元素,若比堆顶元素小,则将其加入堆中,同时将堆顶元素删除,最终堆中的k个元素即为所求。具体代码如下: import heapq def find_k_min(nums, k): if not nums or k <= 0 or k > len(nums): return None res = [] for i in range(k): heapq.heappush(res, nums[i]) for i in range(k, len(nums)): if nums[i] < res[0]: heapq.heappop(res) heapq.heappush(res, nums[i]) return res 方法二:使用快排的思想 使用快排的思想,每次选择一个基准元素,将小于基准元素的放在左边,大于等于基准元素的放在右边,然后判断基准元素的位置,若该位置等于k-1,则基准元素以及左边的元素就是所求;若该位置小于k-1,则在右半边继续查找;若该位置大于k-1,则在左半边继续查找。具体代码如下: def find_k_min(nums, k): if not nums or k <= 0 or k > len(nums): return None left, right = 0, len(nums)-1 while True: index = partition(nums, left, right) if index == k-1: return nums[:k] elif index < k-1: left = index + 1 else: right = index - 1 def partition(nums, left, right): pivot = nums[left] while left < right: while left < right and nums[right] >= pivot: right -= 1 nums[left] = nums[right] while left < right and nums[left] < pivot: left += 1 nums[right] = nums[left] nums[left] = pivot return left 希望能对您有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值