Leetcode 173. 二叉搜索树迭代器

刚刚拿到这个题目的时候一脸懵逼,这啥子情况。冷静一想,悟出其中奥妙。

不就是搜索二叉树进行中序遍历一下,然后用hasnext和next方法操作一下而已啦。

很简单很简单,Mid方法是自己写的中序遍历的方法。用一个Deque来做容器,虽然STL源码剖析上说这个deque能不用最好不用,因为其复杂度比较高,操作起来比较费事,但是图个方便嘛。下面为源码。

 

 AC解:

class BSTIterator {
public:
	deque 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;
	}
};

 

你可能感兴趣的:(Leetcode,Leetcode)