从函数遍历树类

Traversing a tree class from a function

本文关键字：遍历 函数      更新时间：2023-10-16

我有一个树类-见下文-我想遍历树类外部的函数并进行一些处理。为了遍历树，我需要设置一个指向树根的指针。但是我不能访问类之外的根，因为它是私有的。

有没有一种方法可以在不使用getter（检索根地址）的情况下优雅地实现这一点，并且在不公开根的情况下？

谢谢你的帮助。

template <class Key> class IntervalST
{
private:
    Interval<Key> *root;
    bool isRed(Interval<Key> *interval);
    Interval<Key> *rotateLeft(Interval<Key> *h);
    Interval<Key> *rotateRight(Interval<Key> *h);
    Interval<Key> *put(Interval<Key> *h,Key lo, Key hi, Key val);
    Interval<Key> *moveRedLeft(Interval<Key> *h);
    Interval<Key> *moveRedRight(Interval<Key> *h);
    Interval<Key> *deleteMin(Interval<Key> *h, Key hi);
    Interval<Key> *balance(Interval<Key> *h);
    Interval<Key> *remove(Interval<Key> *h, Key lo, Key hi);
    Interval<Key> *min(Interval<Key> *h);
    Interval<Key> *addDuplicate(Interval<Key> *h, Key hi);
    Interval<Key> *removeDuplicate(Interval<Key> *h, Key low, Key hi);
    Interval<Key> *getPointerToKey(Key low);
    void flipColors(Interval<Key> *h);
    void destroy(Interval<Key> *h);
    void printTree(Interval<Key> *h, int indent);
    Key maxVal(Interval<Key> *h);
    int size(Interval<Key> *h);
    bool isBST(Interval<Key> *x, Key min, Key max);
    inline bool isBST(){return isBST(root,0,0);}
    bool isSizeConsistent(Interval<Key> *x);
    inline bool isSizeConsistent(){return isSizeConsistent(root);}
    bool is23(Interval<Key> *x);
    inline bool is23(){return is23(root);}
    bool isBalanced();
    bool isBalanced(Interval<Key> *x,int black);
    int getKeySize(Key low);
    int compare(Key a, Key b);
public:
    //don't forget to build the constructor
    //and overload the =equal operator
    IntervalST():root(NULL){};
    ~IntervalST();
    void remove(Key lo, Key hi);
    void put(Key lo, Key hi);
    inline int size(){return size(root);}
    inline bool isEmpty(){return root == NULL;}
    void print(int indent = 0);
    void check();
};

您可以向树类添加接受函数指针的方法，这样您就可以向它传递一个函数，该函数将在树的每个节点上执行。这保持了在树中遍历封装，同时允许将指针传递到要在每个节点上执行的任何函数。

像这样：

void IntervalST::executeOnEachNode(void (*functor)(Interval<Key> node,void* userData),
                                   void *userData = 0)
{
      Interval<Key> node;
      //Loop to traverse your tree stepping through each node
     {
       //Calls functor with supplied user data on specific node ( 
       functor(node,userData);
     }
}

其中，functor是指向接受一个参数（void*）的函数的指针-userData，如果需要，可以使用它向函数传递绑定的附加参数。

用例（计算所有树节点）：

void countAllNodes(Interval<Key> node,void* additionalIntArgument)
{
    int* count = static_cast<int*>(additionalIntArgument);
    *count += 1;
}
IntervalST<double> tree;
int count(0);
tree.executeOnEachNode(countAllNodes,reinterpret_cast<void*>(&count));
std::cout << "Tree has "<<count << " nodes "<<std::endl;

您想要的是类的迭代器。它很可能会被定义为一个朋友，这样它就可以看到内部。否则，您需要公开方法进行访问。

定义我自己容器的迭代器以获取详细信息。

相关文章：