刚刚拿到这个题目的时候一脸懵逼,这啥子情况。冷静一想,悟出其中奥妙。
不就是搜索二叉树进行中序遍历一下,然后用hasnext和next方法操作一下而已啦。
很简单很简单,Mid方法是自己写的中序遍历的方法。用一个Deque来做容器,虽然STL源码剖析上说这个deque能不用最好不用,因为其复杂度比较高,操作起来比较费事,但是图个方便嘛。下面为源码。
AC解:
class BSTIterator {
public:
deque<int> Temp;
BSTIterator(TreeNode *root) {
if (root != NULL)
Mid(root);
}
void Mid(TreeNode *root)
{
if (root->left != NULL)
Mid(root->left);
Temp.push_back(root->val);
if (root->right!= NULL)
Mid(root->right);
}
/** @return whether we have a next smallest number */
bool hasNext() {
if (Temp.size() != 0)
return true;
else
return false;
}
/** @return the next smallest number */
int next() {
int Res = Temp[0];
Temp.pop_front();
return Res;
}
};