【初阶数据结构与算法】第八篇——二叉树的顺序结构的应用(堆排序+TOPK问题)

本文详细介绍了二叉树在排序算法中的应用,特别是堆排序的概念、两种实现方式及其时间复杂度分析。此外,还探讨了如何利用堆解决TOP-K问题,通过建立小堆或大堆来寻找数组中最大的K个数,同时给出了相应的代码实现。
摘要由CSDN通过智能技术生成

⭐️本篇博客我要给大家分享一下二叉树顺序结构应用。希望对大家有所帮助。
⭐️ 博主码云gitee链接:码云主页

 💙系列文章💙

【初阶数据结构与算法】第一篇:算法中的时间复杂度和空间复杂度

【初阶数据结构与算法】第二篇:顺序表

【初阶数据结构与算法】第三篇:单链表

 【初阶数据结构与算法】第四篇:链表面试题详解

【初阶数据结构与算法】第五篇:双链表

【初阶数据结构与算法】第六篇:栈和队列(各个功能实现+练习题包含多种方法)

【初阶数据结构与算法】第七篇:二叉树和堆的基本概念+以及堆的实现


目录

 💙系列文章💙

前言

🌏一、堆排序

        🍯1.堆排序概念

         🍯2.堆排序方法实现和代码

                🍍建立堆(以下以建立大堆为例)

                🍍排序选择堆

🌏二、TOP-K问题

🌏总结


前言


🌏一、堆排序

        🍯1.堆排序概念

⭐️堆排序:是通过建立大堆或者小堆来排序的算法

         🍯2.堆排序方法实现和代码

                🍍升序-建立堆-向上调整

⭐️第一种:使用向上调整插入数据的思想建立堆,将第一个数字看成堆,后面的数假装是插入的,然后一个一个向上调整

请添加图片描述

void AjustUp(int* a, size_t child);

void HeapSort(int* a, int size) {
	//向上调整,建堆
	for (int i = 1; i < size; i++) {
		AjustUp(a, i);
	}
}

 ⭐️第二种:向下调整,首先从非叶子节点开始(一个叶子节点既可以看作小堆,也可以看作大堆,不会影响结构,所以从倒数第1个非叶子节点开始),最后一个节点是n-1,那么最后一个非叶子节点就是(n-1-1)/2,从倒数第一个非叶子节点开始向下调整,直到根节点结束

 请添加图片描述

void HeapSort(int* a, int size){
    //最后一个元素的下标是n-1,他的父亲节点下标就是(n-1-1)/2
    int parent = (n - 1 - 1) / 2;
    int i = 0;
    for (i = parent; i >= 0; i--)
    {
    	//建大堆 向下调整
    	AdjustDown(hp->a, n, i);
    }
}

        🍍降序-建立小堆-向下调整  

⭐️建小堆 

现在有一个数组

int arr[] = { 70,56,30,24,33,10,75,68,90,21};

画成二叉树是

 我们从最后一个父亲节(即最后一个叶子节点的父亲)点开始,向下调整,parent = (child-1)/2,最后一个叶子节点是n-1(n为整个数组的长度,n-1是最后一个叶子节点的下标),所以最后一个父亲节点是(n-1-1)/2,然后向下调整,第一波向下调整完成后是下面这样

 之后我们从倒数第二个父亲节点开始,也就是最后一个父亲节点的旁边一个,然后向下调整,之后下一轮从倒是第三个父亲节点开始,向下调整,直到最后到根节点,建小堆完成后是下面这样

 代码:

void CreateHeap(int* a,int size)
{
    //最后一个节点的位置是size-1,则可以得到双亲节点的位置 
    int parent = (size-1-1)/2;
    while(parent >= 0)
    {
        //调整子树
        AdjustDown(a,parent,size);
        
        //调整完以后,parent向前移动
        parent--;
    }
}

⭐️向下调整排序

        每轮排序,需要将第一个元素和最后一个元素交换,将已经得到最小元素放到最后面,然后将剩下的n-1个元素重新调整,如下图,将圈住的n-1个元素重新调整(最后一个元素不参与调整),所以算法总长度这里要-1,然后继续向下调整,调整完后end减1方便选出下一次和root交换的数

 

(1)先把第一个元素和最后一个元素交换

(2)然后需要调整的节点个数size 减一

(3)从根部向下调整

(4)树的尾部 end 向前移动,end 减一

代码

void HeapSort(int* a,int size)
{
	CreateHeap(a, size);
 
 
	int end = size - 1;
	while (end>0)
	{
		//将堆顶元素和堆底元素互换
		Swap(a[0], a[end]);
		//交换以后,最小的数放在堆底,要参加堆调整的节点个数要减一
		size--;
 
		//由于子树都是小堆,直接从堆顶向下调整
		AdjustDown(a, 0, size);
 
		//每得到一个最小的,都会被放到最后,最后一个节点的位置向前移动
		end--;
	}
}

        🍍排序选择堆

🔑升序和降序选择大堆小堆?

⭐️ 升序选大堆,降序选小堆

        升序如果选小堆,那么如果第一个数字是最小的,那么选出第二小的数的时候,就要重新建堆,使后面队形散乱,所以升序要选大堆。相反降序就选小堆。

        将第一个数(即最大的数)和最后一个叶子节点交换(将大的数放到最后-升序)然后把前面n-1个元素当成一个新的堆然后向下调整,得到堆顶元素是次大的数,然后依次交换次大的数和倒数第二个元素

void HeapSort(HPDataType* a, int n)
{
	//找到最后一个父亲节点
	int parent = (n - 1 - 1) / 2;
	int i = 0;
	//建小堆 排降序  建大堆 排升序
	for (i = parent; i >= 0; i--)
	{
		//建大堆 向下调整
		AdjustDown(a, n, i);
	}

	i = n - 1;
	while (i >= 0)
	{
		Swap(&a[0], &a[i]);
		AdjustDown(a, i, 0);
		i--;
	}
}

⭐️时间复杂度是o(n*logn),建立堆时间复杂度是o(n),排序是o(logn)

⭐️空间复杂度是O(1)

🌏二、TOP-K问题

⭐️TOPK问题:找出N个数里面最大/最小的前K个问题。

⭐️思路:用K个数建立K个数的小堆(如果建大堆,最大的数可能挡在头的位置,其他9个次大的数就进不来,用然后剩下的N-K个数依次遍历,如果比堆顶的数据大,就替换他进堆(向下调整),最后堆里面的K个数就是最大的K个

void PrintTopk(int* a, int n, int k) {
	int* kminHeap = (int*)malloc(sizeof(int)*k);
	assert(kminHeap);
	for (int i = 0; i < k; i++) {
		kminHeap[i] = a[i];
	}
	int parent = (k - 1 - 1) / 2;
	for (int i = parent; i >= 0; i--) {
		AjustDown(kminHeap, k, i);
	}
	for (int i = k; i < n; i++) {
		if (a[i] > kminHeap[0]) {
			kminHeap[0] = a[i];
			AjustDown(kminHeap, k, 0);
		}
	}
	for (int i = 0; i < k; i++) {
		printf("%d ", kminHeap[i]);
	}
	printf("\n");
}

void TestTopk()
{
    //随机取十个很大的数
	int n = 100000;
	int* a = (int*)malloc(sizeof(int) * n);
	srand(time(0));
	for (size_t i = 0; i < n; ++i)
	{
		a[i] = rand() % 1000000;
	}
	//随机赋值十个最大的数
	a[5] = 1000000 + 1;
	a[1231] = 1000000 + 2;
	a[531] = 1000000 + 3;
	a[5121] = 1000000 + 4;
	a[115] = 1000000 + 5;
	a[2335] = 1000000 + 6;
	a[9999] = 1000000 + 7;
	a[76] = 1000000 + 8;
	a[423] = 1000000 + 9;
	a[3144] = 1000000 + 10;

	PrintTopK(a, n, 10);
}

 ⭐️首先开辟好十个空间,存放前K个数据,再将这K个数据建立为小堆, 之后将剩下N-K个数据依次向上调整(此时K数组中第一个数据是最小的),然后在向上调整。

⭐️复杂度分析:时间复杂度O(K+logK(N-K)),建堆是O(K),向下调整是(N-K)*logK,当N很大,K很小时,可以等价于O(N)

⭐️空间复杂度是O(K)


🌏总结

        有时候觉得世界上聪明的人还是越多越好,一般人谁会想到这种方式呢

  • 27
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 36
    评论
评论 36
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

penguin_bark

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值