TopK问题_大顶堆实现_C#_适合新手_全注释

参考这个前辈的博客https://blog.csdn.net/luochoudan/article/details/53736752?tdsourcetag=s_pctim_aiomsg，从该前辈代码的基础上修改成了C#版本，要看思路的可以看阅读该博客

原题：服务器有10亿的玩家，如何在排行榜上显示战力前1000的玩家(经典topK问题)

主要是阅读前辈博客时，基础不牢固，导致一些地方有疑惑。多方查阅后，将一些个人理解写在注释中。

using System;

public class TopK
{
    public static void Main(string[] args)
    {
        int[] a = { 1, 17, 4, 3, 5, 6, 7, 16, 9, 10, 11, 12, 13, 14, 15, 8 };
        int[] b = topK(a, 4);
        for (int i = 0; i < b.Length; i++)
        {
            Console.Write(b[i] + ", ");
        }
    }

    public static void heapify(int[] array, int index, int length)
    {
        //按照2*index+1为左节点，2*index+2为右节点的存储方式
        //这种存储方式的直观好处是，堆顶就放在第一个元素
        int left = index * 2 + 1;
        int right = index * 2 + 2;
        int largest = index;
        //这个判断逻辑实现的是大顶堆，若要小顶堆，把array[left] > array[index]改为array[left] < array[index]即可
        if (left < length && array[left] > array[index])
        {
            largest = left;
        }
        //若要实现小顶堆array[left] > array[index]改为array[left] < array[index]即可
        if (right < length && array[right] > array[largest])
        {
            largest = right;
        }
        if (index != largest)
        {
            swap(array, largest, index);
            //每当找到一个异常的节点时，要递归比较其子节点，以不断地将其"沉降"
            heapify(array, largest, length);
        }
    }

    public static void swap(int[] array, int a, int b)
    {
        //使用三个异或，快速的交换数据，且不用生成新的变量
        array[a] = array[a] ^ array[b];
        array[b] = array[a] ^ array[b];
        array[a] = array[a] ^ array[b];
    }

    public static void buildHeap(int[] array)
    {
        int length = array.Length;
        //按照2*index+1为左节点，2*index+2为右节点的存储方式，这里是右左根的遍历顺序
        //这里从length/2-1的原因是，要从叶子节点上开始遍历,PS:要注意length/2会有一个int的取整操作
        //依照2*index+1为左节点，2*index+2为右节点的存储方式,length/2-1后面的值都将是叶子节点。
        //PS:可以验算一下，若length可以整除2，(length/2-1)*2+1=length - 1，刚好是数组的最后一个元素作为左叶子
        //PS:若length不可以整除2，(length/2-1)*2+2=length - 1，刚好是数组的最后一个元素作为右叶子
        for (int i = length / 2 - 1; i >= 0; i--)
        {
            heapify(array, i, length);
        }
    }

    public static void setTop(int[] array, int top)
    {
        array[0] = top;
        heapify(array, 0, array.Length);
    }

    //要注意，这里建的是大顶堆，所以求得是最小的TopK个元素
    public static int[] topK(int[] array, int k)
    {
        int[] top = new int[k];
        for (int i = 0; i < k; i++)
        {
            top[i] = array[i];
        }
        //先建堆，然后依次比较剩余元素与堆顶元素的大小，比堆顶小的，说明它应该在堆中出现，则用它来替换掉堆顶元素，然后沉降。
        buildHeap(top);
        for (int j = k; j < array.Length; j++)
        {
            int temp = top[0];
            //若要实现取最大的TopK个元素，把这里 < 改为 >，然后将上面的大顶堆改为小顶堆即可
            if (array[j] < temp)
            {
                setTop(top, array[j]);
            }
        }
        return top;
    }
}