前序遍历
题目来源:https://leetcode.cn/problems/binary-tree-preorder-traversal/description/
通过栈进行遍历,也就是访问元素后将其入栈,然后访问右节点,再访问左节点(出栈顺序相反),代码如下:
class Solution{
public:
vector<int> preorderTraversal(TreeNode* root){
vector<int> result;
stack<TreeNode*> st;
if (root == nullptr) return result;
st.push(root);
while(!st.empty()){
TreeNode * node = st.top();
st.pop();
result.push_back(node->val);
if(node -> right) st.push(node -> right);
//注意这里要将节点放入栈,而不是节点的数值!
if(node -> left) st.push(node -> left);
}return result;
}
};
举个例子来论证过程:给出一个二叉树[5,4,1,-1,-1,2,6,-1,-1],前序遍历顺序是中左右,也就是先遍历中介点,然后去左节点,右节点,顺序是54126,程序思路是做一个做一个栈来统计出入的值,做一个一维数组result存放遍历结果。首先把根节点入栈,然后判断根节点的值放入result,再求它右节点和左节点,此时把6和4按先后进栈,这个时候node为4,然后4出栈,完后处理思路和前面一样。
注意点:
1、栈类型是TreeNode :
stack<typename> name;
//这里的typename可以是任何基本类型,例如 int、double、char、结构体等,也可以是STL标准容器,例如vector、set、queue等。
中序遍历:
和递归法不同,迭代法的中序遍历并不是前序遍历修改顺序,原因很简单,前序遍历的顺序是中左右,先遍历中节点然后是左右,正好我们入栈顺序也是中节点先进去,无非是左右节点交换进去的顺序以保证出栈的顺序相反。即要访问的元素和要处理的元素顺序是一致的,都是中间节点。但是中序遍历要求左中右,也就是先遍历左边的节点,然后中间节点,右节点。这个时候带来了问题,要访问的元素和要处理的元素顺序不一致。
那么再看看中序遍历,中序遍历是左中右,先访问的是二叉树顶部的节点,然后一层一层向下访问,直到到达树左面的最底部,再开始处理节点(也就是在把节点的数值放进result数组中),这就造成了处理顺序和访问顺序是不一致的。——代码随想录。
那么在使用迭代法写中序遍历,就需要借用指针的遍历来帮助访问节点,栈则用来处理节点上的元素。
代码如下,共性的解释见前序遍历迭代法。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;//一维数组,用于存放遍历的节点数值
stack<TreeNode*> st;
TreeNode* cur = root;//建立指针,用于访问节点(指针访问的节点存到st中,和result没关系)
while (cur != NULL || !st.empty()) {
if (cur != NULL) { // 指针来访问节点,访问到最底层
st.push(cur); // 将访问的节点放进栈
cur = cur->left; // 左
} else {
cur = st.top(); // 这个时候cur指向的是空,所以我们从栈顶的第一个值赋予cur,从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
st.pop();
result.push_back(cur->val); // 中
cur = cur->right; // 右
}
}
return result;
}
};
通俗的语言来解释,那就是:给出一个二叉树[5,4,1,-1,-1,2,6,-1,-1]通过指针一路向左遍历同时入栈st,直到发现节点为空,也就是遍历到了二叉树最左边,这个时候指针cur指向了1的左孩子,此时节点为空,此时指针指向1,1出栈并存入result,指针指向1的右孩子,此时还是空,则再次出栈,指针指向出栈的数值所在节点.....
也就是一直往左遍历,遇到空就处理出栈中节点,然后往右,如果右节点不为空,那就继续往左走,为空则继续出栈。
后序遍历:
先序遍历是中左右,后续遍历是左右中,其中左右都是绑定在一起的,而不是中序的走中右。那么我们只需要调整一下先序遍历的代码顺序,就变成中右左的遍历顺序,然后在反转result数组,输出的结果顺序就是左右中了,如下图:
class Solution{
public:
vector<int> preorderTraversal(TreeNode* root){
vector<int> result;
stack<TreeNode*> st;
if (root == nullptr) return result;
st.push(root);
while(!st.empty()){
TreeNode * node = st.top();
st.pop();
result.push_back(node->val);
//if(node -> right) st.push(node -> right);
if(node -> left) st.push(node -> left);
if(node -> right) st.push(node -> right);
}
reverse(result.begin(), result.end()); // 结果反转
return result;
}
};