LeetCode
来源:互联网 发布:王者荣耀输入法软件 编辑:程序博客网 时间:2024/06/04 23:31
Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.
Calling next()
will return the next smallest number in the BST.
Note: next()
and hasNext()
should run in average O(1) time and uses O(h) memory, where h is the height of the tree.
构造一个迭代器,每次查询二叉搜索树中下一个最小值。
用一个栈来构造,每次把一个点的所有左子树丢进去。next()可以实现O(1)time,hasNext()是O(h)time。空间复杂度O(h)
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class BSTIterator {public: stack<TreeNode*> st; BSTIterator(TreeNode *root) { solve_left(root); } /** @return whether we have a next smallest number */ bool hasNext() { return !st.empty(); } /** @return the next smallest number */ int next() { TreeNode* cur = st.top(); st.pop(); if (cur->right) { solve_left(cur->right); } return cur->val; } void solve_left(TreeNode* root) { while (root) { st.push(root); root = root->left; } return; }};/** * Your BSTIterator will be called like this: * BSTIterator i = BSTIterator(root); * while (i.hasNext()) cout << i.next(); */
阅读全文
0 0
- leetcode
- [leetcode]
- LeetCode
- leetcode
- leetcode
- leetcode:
- leetcode:
- LeetCode
- leetcode
- LEETCODE
- leetcode
- leetCode
- leetcode
- [leetcode]
- LeetCode
- leetcode
- leetcode:
- leetcode
- python 获取图片分辨率
- OpenCV之图形学的腐蚀与膨胀
- 多态和多态对象模型
- 一个求数组中最大元素的函数模板
- linux之Vim文档编辑
- LeetCode
- HDU
- 网络编程基础集
- 【Java 你应该知道的】 泛型01
- html5 css js轮播图动画
- Fluent FIDAP v8.7_简体中文教程7.rar
- ASP.net button验证
- String,StringBuffer,StringBuilder的区别
- 编写shell脚本遇到的问题