浙大数据结构:05-树7 堆中的路径

这道题主要难在创建一个小根堆,输出并不难,而创建代码可以参考MOOC上代码。

1、条件准备

定义结构体,data数组来存储堆的,size是当前堆的大小,MaxSize是堆的最大大小。
cpp 复制代码
#include <iostream>
using namespace std;

typedef int ElementType;
typedef int position;
typedef struct HNode   
{
  ElementType *data;
  int size ;
  int MaxSize;
}* Heap;

typedef Heap MinHeap;

#define MIN -20000
主函数先是读入结点数量,输出组数。
然后初始化一下堆,输入元素建立小根堆。
然后循环调用solve函数输出每一组的数据。
f是用来控制输出格式的保证没有多余空格。
cpp 复制代码
int main()
{
  ios::sync_with_stdio(false);
  cin.tie(0), cout.tie(0);
  int n, l;
  cin >> n >> l;
  MinHeap H = createheap(5);
  for (int i = 0; i < n; i++)
  {
    int elem;
    cin >> elem;
    insert(H, elem);
  }
  int f = 0;
  for (int i = 0; i < l; i++)
  {
    if (f)
      cout << endl;
    int node;
    cin >> node;
    solve(H, node);
    f = 1;
  }

  return 0;
}

2、createheap函数

分配空间,一般多分配一点。
初始化里面数据,data0作为哨兵结点。
cpp 复制代码
MinHeap createheap(int maxsize)
{
  MinHeap H = (MinHeap)malloc(sizeof(struct HNode));
  H->data = (ElementType *)malloc(sizeof(ElementType) * (maxsize + 5));
  H->size = 0;
  H->MaxSize = maxsize;
  H->data[0] = MIN;
  return H;
}

3、insert函数

将x插入到堆末尾,然后循环与其父亲结点比较,比它小就交换,直到不小于。
这里利用完全二叉树的性质:如果根节点标为1的话,则树中每一个结他的左节点为2i,右结点为2i+1。
cpp 复制代码
bool insert(MinHeap H, ElementType x)
{
  int i;
  i = ++H->size;
  for (; H->data[i / 2] > x; i /= 2)
  {
    H->data[i] = H->data[i / 2];
  }
  H->data[i] = x;
  return true;
}

4、solve函数

传入起始位置,然后一直输出,下一次的位置为当前位置的父亲结点的位置。
f是为了控制没有多余空格输出。
cpp 复制代码
void solve(MinHeap H, position x)
{
  int f = 1;
  for (; x; x /= 2)
  {
    if (f)
    {
      cout << H->data[x];
      f = 0;
    }
    else
      cout << ' ' << H->data[x];
  }
}

5、总结

该题主要考察堆的相关性质,以及完全二叉树的性质,难度不算大。
完整代码如下:
cpp 复制代码
#include <iostream>
using namespace std;

typedef int ElementType;
typedef int position;
typedef struct HNode
{
  ElementType *data;
  int size;
  int MaxSize;
} *Heap;

typedef Heap MinHeap;

#define MIN -20000

MinHeap createheap(int maxsize)
{
  MinHeap H = (MinHeap)malloc(sizeof(struct HNode));
  H->data = (ElementType *)malloc(sizeof(ElementType) * (maxsize + 5));
  H->size = 0;
  H->MaxSize = maxsize;
  H->data[0] = MIN;
  return H;
}

bool insert(MinHeap H, ElementType x)
{
  int i;
  i = ++H->size;
  for (; H->data[i / 2] > x; i /= 2)
  {
    H->data[i] = H->data[i / 2];
  }
  H->data[i] = x;
  return true;
}

void solve(MinHeap H, position x)
{
  int f = 1;
  for (; x; x /= 2)
  {
    if (f)
    {
      cout << H->data[x];
      f = 0;
    }
    else
      cout << ' ' << H->data[x];
  }
}

int main()
{
  ios::sync_with_stdio(false);
  cin.tie(0), cout.tie(0);
  int n, l;
  cin >> n >> l;
  MinHeap H = createheap(5);
  for (int i = 0; i < n; i++)
  {
    int elem;
    cin >> elem;
    insert(H, elem);
  }
  int f = 0;
  for (int i = 0; i < l; i++)
  {
    if (f)
      cout << endl;
    int node;
    cin >> node;
    solve(H, node);
    f = 1;
  }

  return 0;
}

附录:堆的定义与操作

cpp 复制代码
#include <iostream>
using namespace std;

typedef int ElementType;

typedef struct HNode   
{
  ElementType *data;
  int size ;
  int MaxSize;
}* Heap;

typedef Heap MaxHeap;
typedef Heap MinHeap;

#define MAX 1000

MaxHeap createheap(int maxsize)
{
  MaxHeap H=(MaxHeap)malloc(sizeof(struct HNode));
  H->data=(ElementType*)malloc(sizeof(ElementType)*(maxsize+1));
  H->size=0;
  H->MaxSize=maxsize;
  H->data[0]=MAX;
  return H;
}

bool isfull (MaxHeap H)
{
  return (H->size==H->MaxSize);
}
bool insert(MaxHeap H ,ElementType x)
{
  int i;
  if(isfull(H))return false;

  i=++H->size;
  for(;H->data[i/2]<x;i/=2)
  {
    H->data[i]=H->data[i/2];
  }
  H->data[i]=x;
  return true;

}

#define ERROR -1

bool isempty(MaxHeap H)
{
  return H->size==0;
}

ElementType deletemax(MaxHeap H)
{
 int parent ,child;
 ElementType maxitem,x;
 if(isempty(H))return ERROR;

 maxitem=H->data[1];
 x=H->data[H->size--];

 for(parent=1;parent*2<=H->size;parent=child)
 {
  child=parent*2;
  if(child!=H->size&&H->data[child]<H->data[child+1])
  child++;
  if(x>=H->data[child])break;
    else H->data[parent]=H->data[child];
 }

 H->data[parent]=x;
  return maxitem;

}

void  percdown(MaxHeap H,int p)
{
  int parent,child;
  ElementType x;

  x=H->data[p];
for(parent=p;parent*2<=H->size;parent=child)
{
  child=parent*2;
  if(child!=H->size&&H->data[child]<H->data[child+1])
   child++;
  if(x>=H->data[child])break;
  else H->data[parent]=H->data[child];
}
H->data[parent]=x;

} 

void buildheap(MaxHeap H)
{
  int i;
  for(i=H->size/2;i>0;i--)
  {
    percdown(H,i);
  }
}
相关推荐
wuyk5551 分钟前
4.树:一对多的层次数据结构
开发语言·数据结构·stm32·单片机
en.en..4 小时前
数据结构考点解析:逻辑闭环与存储循环辨析
数据结构
布莱克6056 小时前
数据库索引分类:数据结构、物理存储与逻辑角度详解
数据结构·数据库
Fa_Mian_Tuan7 小时前
图论基础|邻接矩阵超详细讲解(含无向/有向/带权图+完整可运行C语言代码)
c语言·数据结构·笔记·算法·图论
小星星闪亮登场9 小时前
ST表--倍增思想
开发语言·数据结构·c++·算法·思维
xiaohebang10 小时前
流失预测模型设计:行为特征分析算法选型
大数据·数据结构·经验分享
sunoo-22910 小时前
数据结构入门 Day3:双向链表与双向循环链表全解析(核心操作 + 易错坑点 + 三类链表对比)
linux·开发语言·数据结构·笔记·学习·链表
祖力5512 小时前
快速学会数据结构中的链式队列与循环队列
linux·c语言·数据结构·算法
布莱克60513 小时前
理解B+树:原理、特性与应用场景
数据结构·数据库·mysql
晚风叙码15 小时前
C++哈希表实现:开放定址法和链地址法 (哈希桶)
数据结构·c++·哈希算法·散列表