Я пытаюсь реализовать деревья сегментов для извлечения минимального значения из заданного интервала в массиве. 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 и индекс 11 массива, представляющего дерево сегментов, показывают значение nan
; в то время как ничего подобного не происходит в случае 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
Может кто-нибудь указать причину этого?
Функция constructST
никогда ничего не возвращает. Если вы обновите это, чтобы вернуться tree[currentIndex]
вы перестанете получать NaNs в ваших результатах. Тот факт, что он работал с 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));
return tree[currentIndex];
}
Других решений пока нет …