【发布时间】:2017-11-29 04:48:06
【问题描述】:
我只是尝试构建一个最小堆,但我得到了构建堆的不同答案
方法 1 将元素插入数组,然后调用构建 minheap 方法,该方法将 minHeapify 应用于内部节点。
方法 2 通过检查数组是否遵循 minheap 属性,将元素直接插入堆中。
我猜两个答案都是正确的,但是如果某些测试用例显示一个订单并且答案不同,那可以吗?
#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()) 产生与方法 1 相同的 ans 和 min_heap(b.begin(),b.end(),myComparator()) 产生相同的 ans 产生通过方法2。所以我只是想确认一下这件事好吗???
【问题讨论】:
-
在编码竞赛中,如果测试用例有预定义的输出,我怎么知道要使用哪种方法?
标签: c++ binary-heap