我只是尝试构建一个最小堆,我得到了不同的答案来构建堆
方法1将元素插入到数组中,然后调用构建minheap方法,该方法在内部节点上应用minHeapify。
方法2通过在每个点检查数组是否遵循minheap属性,将元素直接插入堆中。
我猜这两个答案都是正确的,但是如果某个测试用例显示一个订单并且ans不同则可以。
#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
int leftChild(int i)
{
return (2*i+1);
}
int rightChild(int i)
{
return (2*i+2);
}
int parent(int i)
{
return ((i-1)/2);
}
void minHeapify(vector<int> &a,int i)
{
int smallest = i;
int l = leftChild(i);
int r = rightChild(i);
if(l < a.size() && a[l] < a[i])
smallest = l;
if(r < a.size() && a[r] < a[smallest])
smallest = r;
if(smallest != i)
{
swap(a[smallest],a[i]);
minHeapify(a,smallest);
}
}
void buildMinHeap(vector<int> &a)
{
for(int i = a.size()/2 - 1 ; i >= 0 ; i--)
minHeapify(a,i);
}
void insertKey(vector<int> &b,int new_val,int* n)
{
*n = *n + 1;
int i = *n - 1;
b[i] = new_val;
while(i!=0 && b[i] < b[parent(i)])
{
swap(b[i],b[parent(i)]);
i = parent(i);
}
}
void printline()
{
cout<<"********************************************************************"<<endl;
}
int main()
{
cout<<"Enter the elements in the array for building a min heap"<<endl;
vector<int> a;
a.push_back(2);
a.push_back(16);
a.push_back(74);
a.push_back(58);
a.push_back(36);
a.push_back(4);
a.push_back(28);
a.push_back(15);
a.push_back(35);
a.push_back(82);
a.push_back(6);
//One thing you can do is make a normal array and build a heap out of it in O(n) time
buildMinHeap(a);
for(int i=0;i<a.size();i++)
cout<<a[i]<<" ";
cout<<endl<<endl;
printline();
//Or second method is insert in a way such that it is always inserted in a form of heap.
vector<int> b(10000);
int heapsize = 0;
insertKey(b,2,&heapsize);
insertKey(b,16,&heapsize);
insertKey(b,74,&heapsize);
insertKey(b,58,&heapsize);
insertKey(b,36,&heapsize);
insertKey(b,4,&heapsize);
insertKey(b,28,&heapsize);
insertKey(b,15,&heapsize);
insertKey(b,35,&heapsize);
insertKey(b,82,&heapsize);
insertKey(b,6,&heapsize);
for(int i=0;i<heapsize;i++)
cout<<b[i]<<" ";
}
我使用min_heap函数检查并从我的答案中构造了一个最小堆。 min_heap(a.begin(),a.end(),myComparator())产生与method1和min_heap产生的相同的ans(b.begin(),b.end(),myComparator())产生相同的ans产生通过method2。所以我只想确认这件事好吗???
答案 0 :(得分:0)
对于Heap,保存在基础数据结构中的元素顺序并不重要。
您应该检查Min Heap中的弹出值是否按升序排列,即根元素始终是最小值。
将top()
和pop()
方法添加到您的结构中。 top()
将只返回根元素值,pop()
将删除根元素并将其替换为堆中合适的元素。
以下代码应按相同顺序打印元素。
while(heap.size() > 0)
{
std::cout<<heap.top()<<" ";
heap.pop();
}
对于编码竞赛,他们永远不会检查底层结构的顺序。你应该按顺序打印它们。