前面用迭代法,难以实现统一的代码,不像递归法改一下节点顺序就可以转化成另一个。
统一的迭代遍历实现方法:
以中序遍历为例,由于无法同时解决访问节点(遍历节点)和处理节点(将元素放进结果集)不一致的情况。那么我们就将访问的节点放入栈中,把要处理的节点也放进栈中,但是要做好标记。
标记方式即,在把要处理的节点放入栈后,紧接着放空指针作为标记。这也叫标记法。
class Solution{
public:
vector<int>inorderTraversal(TreeNode* root){
vector<int>result;
stack<TreeNode*>st;
if(root != NULL)st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
if(node != NULL){
st.pop(); //将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
if(node->right)st.push(node->right);//添加右节点(空节点不入栈)
st.push(node); //添加中节点
st.push(NULL); //中节点访问过,但是还没有处理,加入空节点作为标记
if(node->left)st.push(node->left); //添加左节点(空节点不入栈)
}else{ //只有遇到空节点的时候,才将下一个节点放进结果集
st.pop();//将空节点弹出
node = st.top(); //重新取出栈中元素
st.pop();
result.push_back(node->val);//加入到结果集
}
}
return result;
}
}
我们将访问的节点直接加入到栈中,但如果是处理的节点则后面放入一个空节点,这样只有空节点弹出的时候,才将下一个节点放进结果集。
迭代法前序遍历
class Solution{
public:
vector<int>preordderTraversal(TreeNode* root){
vector<int>result;
stack<TreeNode*>st;
if(root != NULL)st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
if(node != NULL){
st.pop();
if(node->right)st.push(node->right);
if(node->lleft)st.push(node->left);
st.push(node);
st.push(NULL);
}else{
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
}
迭代法后序遍历:
class Solution{
public:
vector<int>postorderTraversal(TreeNode* root){
vector<int>result;
stack<TreeNode*>st;
if(root != NULL)st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
if(node != NULL){
st.pop();
st.push(node);
st.push(NULL);
if(node->right)st.push(node->right);
if(node->left)st.push(node->left);
}else{
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};