通过insertKey和buildHeap

时间:2017-06-26 10:55:25

标签: c++ binary-heap

我只是尝试构建一个最小堆,我得到了不同的答案来构建堆

方法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。所以我只想确认这件事好吗???

1 个答案:

答案 0 :(得分:0)

对于Heap,保存在基础数据结构中的元素顺序并不重要。

您应该检查Min Heap中的弹出值是否按升序排列,即根元素始终是最小值。

top()pop()方法添加到您的结构中。 top()将只返回根元素值,pop()将删除根元素并将其替换为堆中合适的元素。

以下代码应按相同顺序打印元素。

while(heap.size() > 0)
{
std::cout<<heap.top()<<" ";
heap.pop();
}

对于编码竞赛,他们永远不会检查底层结构的顺序。你应该按顺序打印它们。