BSTIterator Class Reference

Collaboration diagram for BSTIterator:

Public Member Functions
	BSTIterator (TreeNode *root)
	~BSTIterator ()
int	next ()
bool	hasNext ()

Data Fields
vector< TreeNode * > *	node_array
int	current_index

Detailed Description

Definition at line 20 of file main.cpp.

Constructor & Destructor Documentation

BSTIterator()

BSTIterator::BSTIterator ( TreeNode *  root )

inline

Definition at line 26 of file main.cpp.

                                : node_array(NULL), current_index(0)
    {
        stack<TreeNode*> node_stack;
        TreeNode* current = root;
        node_array = new vector<TreeNode*>{};
 
        while (current != nullptr || !node_stack.empty())
        {
            while (current != nullptr)
            {
                node_stack.push(current);
                current = current->left;
            }
            current = node_stack.top();
            node_stack.pop();
            node_array->push_back(current);
            current = current->right;
        }
    }

References TreeNode::left, node_array, and TreeNode::right.

~BSTIterator()

BSTIterator::~BSTIterator ( )

inline

Definition at line 46 of file main.cpp.

    {
        if (node_array)
            delete node_array;
    }

References node_array.

Member Function Documentation

hasNext()

bool BSTIterator::hasNext ( )

inline

Definition at line 64 of file main.cpp.

    {
        return current_index < (int)node_array->size();
    }

References current_index, and node_array.

Referenced by main().

next()

int BSTIterator::next ( )

inline

Definition at line 52 of file main.cpp.

    {
        if (!node_array)
            return -1; // no array code
        int length = (int)node_array->size();
        if (length <= 0)
            return -2; // empty array code
        if (current_index >= length)
            current_index = 0; // rewind to beginning if domain is exceeded
        return (*node_array)[current_index++]->val; // return current value and increment
    }

References current_index, and node_array.

Referenced by main().

Field Documentation

current_index

int BSTIterator::current_index

Definition at line 24 of file main.cpp.

Referenced by hasNext(), and next().

node_array

vector<TreeNode*>* BSTIterator::node_array

Definition at line 23 of file main.cpp.

Referenced by BSTIterator(), hasNext(), next(), and ~BSTIterator().

---

The documentation for this class was generated from the following file:

• main.cpp