我正在尝试实现段树以从数组中的给定间隔中提取最小值。[参考 - http://www.geeksforgeeks.org/segment-tree-set-1-range-minimum-query/ ]
以下是相同的代码。
//This code inplements segment trees for extracting min from a given interval of an array
#include "iostream"
#include "vector"
#include "cmath"
#include "climits"
using namespace std;
class segmentTree
{
private:
int /* $$$ float $$$ */ *tree;
int n,h; //h represents height of tree
std::vector<int /* $$$ float $$$ */> timeForStick; //vector to store the input array first
public:
segmentTree(int noOfNodes)
{
n=noOfNodes;
h=(int)(ceil(log2(n)));
tree = new int /* $$$ float $$$ */ [(int)(2*pow(2,h))-1];
}
int getMid(int segmentStart, int segmentEnd)
{
return segmentStart+(segmentEnd-segmentStart)/2;
}
int /* $$$ float $$$ */ getMin(int/* $$$ float $$$ */ a, int/* $$$ float $$$ */ b)
{
return (a>b)? b: a;
}
void getInput(int n)
{
for (int i = 0; i < n; ++i)
{
float /* $$$ float $$$ */ temp;
cin>>temp;
timeForStick.push_back(temp); //the input array is stored in a vector timeForStick
}
}
int /* $$$ float $$$ */ constructST(int segmentStart, int segmentEnd, int currentIndex)
{
if (segmentEnd==segmentStart)
{
tree[currentIndex]=timeForStick[segmentEnd];
return timeForStick[segmentEnd];
}
int mid=getMid(segmentStart,segmentEnd);
tree[currentIndex]=getMin(constructST(segmentStart,mid,2*currentIndex+1),constructST(mid+1,segmentEnd,2*currentIndex+2));
}
void printSegmentTreeArray()
{
for (int i = 0; i < (int)(2*pow(2,h))-1; ++i)
{
cout<<i<<"="<<tree[i]<<endl; //print the value at each node(array element) along with indes
}
cout<<"-----------------"<<endl;
}
};
int main(int argc, char const *argv[])
{
int n,l,r;
float min;
cout<<"Enter n";
cin>>n;
segmentTree st(n);
st.getInput(n); //function to get input array from user
st.constructST(0,n-1,0);
st.printSegmentTreeArray();
cin.get();
cin.get();
return 0;
}
如果我正在构建一int
棵树(即使用整数数组存储段树的节点),一切正常。但是一旦我将类型更改为float
,就会发生一些恶作剧,表示段树的数组的索引 0-7 和索引 11nan
显示了值;int
而在树的情况下不会发生这种情况。该函数printSegmentTreeArray()
用于打印存储在表示树的数组的每个位置的值。
笔记:-
1.在代码中,正确更改树的数据类型所需的更改由int
要替换的前面的float
注释块指示。/* $$$ float $$$ */
int
2.我确实编写了从查询间隔中提取 Min 的函数,但由于该constructST()
函数的行为很奇怪,所以将其删除。
3.我已经针对这两种情况使用以下输入测试了我的代码。
18
3 4 2 1 5 7 9 7 10 5 12 3 1 1 2 1 3 2
有人可以指出原因吗?