实现分段树的问题

时间:2013-05-10 16:19:52

标签: c++ data-structures segment-tree

我正在尝试实现Segment Trees以从数组中的给定间隔中提取min值。[参考 - 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 显示值 {{1} } ;虽然在nan树的情况下没有发生这种情况。函数int用于打印存储在表示树的数组的每个位置的值。

注: -

1.在代码中,正确地将树的数据类型从printSegmentTreeArray()更改为int所需的更改由{{1}前面的注释块float指示要被替换。

2.我确实编写了从查询间隔中提取Min的函数,但是由于/* $$$ float $$$ */函数表现得很奇怪而将其删除了。

3.我已使用以下输入测试了我的代码。

int

有人可以指出原因吗?

1 个答案:

答案 0 :(得分:1)

函数constructST永远不会返回任何内容。如果您更新它以返回tree[currentIndex],您将停止在结果中获得NaN。它与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];
}