中序遍历的定义是:首先中序遍历左子树,接着遍历根节点,最后中序遍历右子树
根据这个思路我们就可以试着写出非递归版本的程序。所用的数据结构自然就是栈,首先将根节点入栈,当把左子树中序遍历完了以后,把根节点出栈,再中序遍历右子树。
我们定义节点出栈的过程就是遍历该节点,初始状态我们先让根节点入栈。
由于我们首先遍历的是左子树,所以我们首先不停地进行栈顶元素左孩子入栈操作,直到栈顶元素没有左孩子。接着让栈顶元素出栈,直到出栈元素的右孩子不为NULL,此时说明出栈元素为子树的根节点,接下来需要遍历右子树,所以出栈元素的右子树入栈,重复之前的操作即可。
下面是我的求解程序
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
#include<stack>
#include<vector>
using namespace std;
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> ret;
if(root==NULL) return ret;
st.push(root);
//cout<<"in stack:"<<root->val<<endl;
while(!st.empty()){
while(st.top()->left!=NULL) {
//cout<<"in stack:"<<st.top()->left->val<<endl;
st.push(st.top()->left);
}
while(!st.empty() && st.top()->right==NULL){
ret.push_back(st.top()->val);
//cout<<"out stack:"<<st.top()->val<<endl;
st.pop();
}
if(st.empty()) break;
TreeNode *p=st.top()->right;
//cout<<"out stack:"<<st.top()->val<<endl;
ret.push_back(st.top()->val);
st.pop();
st.push(p);
//cout<<"in stack:"<<p->val<<endl;
}
return ret;
}
};
Runtime: 4 ms, faster than 100.00% of C++ online submissions for Binary Tree Inorder Traversal.
Memory Usage: 9.3 MB, less than 53.58% of C++ online submissions for Binary Tree Inorder Traversal.