堆排序-TOPK问题

现在有n个数，设计算法得到前k大的数。（k<n）

首先从原列表截取前K个元素，组成一个新列表。

然后将这个新列表构建成一个小根堆，那么根就新列表中最小的数。

接着继续从原列表取出K以后的元素，和小根堆根比对，如果比根小，那么肯定不属于前K大的数，如果比根大，替换根，然后做一次向下调整，根重新变成新列表最小的数，接着继续从原列表取数比对，直到取到原列表最后。

那么就可以得到一个前K大的数字组成的小根堆 ，最后出数

先上一下 成果 

 

代码如下 

    public void TOPK(List<int> li,int k)
    {
        PrintList(li,"：初始列表");
        List<int> ki = new List<int>(k);//截取K个元素组成新数组
        if (k<li.Count)
        {
            for (int i = 0; i < k; i++)
            {
                ki.Add(li[i]);
            }
        }
        PrintList(ki,"：前"+k+"个元素组成的列表");
        int n = ki.Count;//新列表长度
        for (int i = ((n - 1 - 1) / 2); i > -1; i--)//从新列表最后一个元素的父节点开始向上调整
        {
            Sift(ki, i, n - 1, false);//最上层就是i，最下层一直指向最后一个元素
        }//得到一个小根堆
        PrintList(ki,"：小根堆");
        for (int i = k; i < li.Count; i++)//从K开始，从原列表取元素比对
        {
            if (li[i] > ki[0])
            {
                ki[0] = li[i];
                Sift(ki, 0, n - 1, false);//每次替换后需要重新向下调整
            }
        }
        PrintList(ki,"：比对完成后的小根堆");
        for (int i = n - 1; i > -1; i--)//开始出数，i指向堆的最后一个元素
        {
            int tmp = ki[i];//交换根和最后一个元素
            ki[i] = ki[0];
            ki[0] = tmp;
            Sift(ki, 0, i - 1, false);//交换结束后向下调整，注意i需要往前移
        }
        PrintList(ki, "：前" + k + "个最大的数");
    }

 

来源：CSDN

作者：lvcoc

链接：https://blog.csdn.net/lvcoc/article/details/104133376