二叉树的遍历主要有两种方式,即深度优先搜索和广度优先搜索,其中深度优先搜索分为前序遍历、中序遍历和后续遍历。每种搜索方式可以掌握递归和迭代两种方式。
144.二叉树的前序遍历
前序遍历
在迭代前序遍历时,需要注意压栈的时机,即先中,再右,再左,这样才可以保证前序遍历中左右的方式
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if (root == NULL) 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;
}
};
func preorderTraversal(root *TreeNode) []int {
ans := []int{}
if root == nil {
return ans
}
st := list.New()
st.PushBack(root)
for st.Len() > 0 {
node := st.Remove(st.Back()).(*TreeNode)
ans = append(ans, node.Val)
if node.Right != nil {
st.PushBack(node.Right)
}
if node.Left != nil {
st.PushBack(node.Left)
}
}
return ans
}
145.二叉树的后序遍历
后序遍历
这道题目在递归中如果使用go语言可以使用闭包函数来写,但是由于闭包函数可以获取函数外的变量,所有递归前必须先声明函数类型,即var xxx func,否则在编译阶段编译器找不到对应的函数,会报undefine。
至于后序遍历的迭代方式,由于只需要将前序遍历时候的先右再左换一下,这样就变成了中、右、左的插入方式,然后进行调换,就成了后序遍历中的左右中遍历方式。
func postorderTraversal(root *TreeNode) []int {
ret := make([]int , 0)
var back func(node *TreeNode)
back = func (root *TreeNode){
if root == nil{
return
}
back(root.Left)
back(root.Right)
ret = append(ret, root.Val)
}
back(root)
return ret
}
94.二叉树的中序遍历
中序遍历的迭代方式是最困难的,因为无法直接拿到中节点来进行引出下一个节点。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
TreeNode* cur = root;
while (cur != NULL || !st.empty()) {
if (cur != NULL) { // 指针来访问节点,访问到最底层
st.push(cur); // 将访问的节点放进栈
cur = cur->left; // 左
} else {
cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
st.pop();
result.push_back(cur->val); // 中
cur = cur->right; // 右
}
}
return result;
}
};