C# 小根堆

堆排序是一个完全二叉树
在这里插入图片描述(来自百度百科)
同时这个完全二叉树还有这一个性质:
根节点总比两个子节点要小(或者要大)
小的叫做小根堆
大的叫做大根堆

逻辑上是一颗树
然后实际使用时,我们通过一个数组对其存储,通过从上而下,从左往右的进行遍历
每个父节点和子节点有如下对应关系:
在这里插入图片描述
相对的:
假设子节点为k
则父节点为:(k+1)/2-1
(偶数int+1之后再除以2不受影响)

由于之前做的那道题目用到了小根堆,所以就以小根堆为例

主要操作为
1.插入
2.移除

插入时,会将值放在数组的末尾,然后从末尾出发,如果这个值所在位置比根节点的值更小,则对其进行交换,直至确认这个值所处位置满足小根堆的特点
(这用到上面的子节点为k,父节点为(k+1)/2-1这个式子)

移除时,将根节点和最后一个值交换,同时移除最后一个值
(网上好像挺多人的做法是用一个array进行存储,然后length表示长度,交换之后length-1,这样方便恢复重建原先的小根堆,length+1然后从最后一个位置出发确定位置,相当于上面的插入那一步了)
移除最后一个值后,再从根节点出发,确认这个值的位置
假如右子树存在时,将这两个子节点中较小的一个和根交换
(用到了根节点为i,子节点分别为i2+1和i2+2)

leetCode题目如下:
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。

示例:

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

返回 13。

说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

        public int KthSmallest(int[][] matrix, int k)
        {
            //int rank = 1;
            int n = matrix.Length;
            minHeap h = new minHeap();
            for (int i = 0; i < n; i++)
                for (int j = 0; j < n; j++)
                {
                    h.insert(matrix[i][j]);
                }


            return h.Sort(k);
        }

        public class minHeap
        {
            private List<int> l = new List<int>();
            public void insert(int val)
            {
                l.Add(val);
                int index = l.Count - 1;
                while ((index + 1) / 2 - 1 >= 0 && l[(index + 1) / 2 - 1] > l[index])
                {
                    int m = l[index];
                    swap(index, (index + 1) / 2 - 1);
                    index = (index + 1) / 2 - 1;
                }
            }
            public int Sort(int k)
            {
                int res = 0;
                for (int i = 0; i < k; i++)
                {
                    res = RemoveMin();
                }
                return res;
            }
            public int RemoveMin()
            {
                int min = l[0];
                int temp = l[l.Count - 1];
                l[0] = temp;
                l.RemoveAt(l.Count - 1);
                int i = 0;
                for (int k = 2 * i + 1; k < l.Count; k = 2 * k + 1)
                {
                    if (k + 1 < l.Count && l[k] > l[k + 1])
                        k++;
                    if (l[k] < temp)
                    {
                        swap(k, i);
                        i = k;
                    }
                    else
                        break;
                }
                return min;
            }
            private void swap(int index1, int index2)
            {
                int m = l[index1];
                l[index1] = l[index2];
                l[index2] = m;
            }

        }

还有好多可以优化的地方:如将list换成int数组。还有什么时候停止小根堆的构建,因为图省事用了list,并且将所有的值都放进去小根堆了,所以执行效率比较低(但是居然还能击败80%多,说明这道题用C#做的人的确很少)

小根堆插入和移除的过程:
插入:
在这里插入图片描述
移除(排序)
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值