-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBSTIterator.cpp.cpp
More file actions
41 lines (38 loc) · 872 Bytes
/
BSTIterator.cpp.cpp
File metadata and controls
41 lines (38 loc) · 872 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
BSTIterator(TreeNode* root)
{
stack<TreeNode*> inOrder;
TreeNode* temp;
if(root)
{
inOrder.push(root);
while(!inOrder.empty())
{
temp=inOrder.top();
if(temp->left)
{
inOrder.push(temp->left);
temp->left=NULL;
}
else
{
inOrder.pop();
st.push(temp->val);
if(temp->right)
{
inOrder.push(temp->right);
temp->right=NULL;
}
}
}
}
}
/** @return the next smallest number */
int next() {
int ret=st.front();
st.pop();
return ret;
}
/** @return whether we have a next smallest number */
bool hasNext() {
return !st.empty();
}