C++ 最小堆的创建堆中的路径

将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i，打印从H[i]到根结点的路径。

输入格式:

每组测试第1行包含2个正整数N和M(≤1000)，分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。

输出格式:

对输入中给出的每个下标i，在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔，行末不得有多余空格。

输入样例:

5 3
46 23 26 24 10
5 4 3

输出样例:

24 23 10
46 23 10
26 10

 

这里堆使用数组下标从1开始的经典实现方法， 这里创建堆的方式使用逐个插入创建的方法，这种方法是时间复杂度是O(NlogN)。而有一种更直接的方法是直接将数组调整为堆，时间复杂是O(logN)。从叶子节点到根节点使用i/2遍历即可。

#include <iostream>
#include <limits>

// 最小堆
class Heap {
private:
	int* heap;
	int size;
public:
	Heap(int size_):size(0){
		heap = new int[size_ + 1];
		heap[0] = INT_MIN;
	}
	~Heap() {
		delete[] heap;
		heap = NULL;
	}
	void Insert(int X) {
		int i;
		for (i = ++size; heap[i / 2] > X; i = i / 2) {
			heap[i] = heap[i / 2];
		}
		heap[i] = X;               
	}

	void Print(int m) {
		std::cout << heap[m];
		m = m / 2;
		while (m) {
			std::cout <<" "<<heap[m];
			m = m / 2;
		}
		std::cout << std::endl;
			
	}
};

int main()
{
	int n, m, temp;
	std::cin >> n >> m;
	Heap H = Heap(n);
	for (int i = 0; i < n; i++) {
		std::cin >> temp;
		H.Insert(temp);
	}
	for (int i = 0; i < m; i++) {
		std::cin >> temp;
		H.Print(temp);
	}
}