参考LeetCode教程
推荐阅读细谈二叉树前序遍历,中序遍历,后序遍历的迭代与递归实现
与二叉树的前中后和层序遍历详细图解(递归和非递归写法)
部分程序参考https://blog.csdn.net/qq_34201858/article/details/104473016
前序遍历
前序遍历首先访问根节点,然后遍历左子树,最后遍历右子树。
这个遍历顺序有点微妙。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
/* 前序遍历的递归实现 */
class Solution {
public:
void process(TreeNode* root,vector<int> &res){
res.push_back(root->val);
if(root->left) process(root->left,res);
if(root->right) process(root->right,res);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if(root){
process(root,res);
}
return res;
}
};
/* 前序遍历的迭代实现 */
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if(!root) return res;
stack<TreeNode*> nodeStack;
nodeStack.push(root);
while(!nodeStack.empty()){
TreeNode* node = nodeStack.top();
res.push_back(node->val);
nodeStack.pop();
if(node->right) nodeStack.push(node->right);
if(node->left) nodeStack.push(node->left);
}
return res;
}
};
中序遍历
中序遍历是先遍历左子树,然后访问根节点,然后遍历右子树。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
/* 中序遍历的递归实现 */
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
if(root){
process(root,res);
}
return res;
}
void process(TreeNode* root,vector<int> &res){
if(root->left) process(root->left,res);
res.push_back(root->val);
if(root->right) process(root->right,res);
}
};
/* 中序遍历的迭代实现 */
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> nodeStack;
TreeNode* cur = root;
while (cur || !nodeStack.empty()) {//循环终止条件
while (cur) {//找到最左节点,并将该路径上的所有节点都入栈
nodeStack.push(cur);
cur = cur->left;
}
TreeNode* top = nodeStack.top();//栈顶元素为最左节点,对其进行访问
res.push_back(top->val);
nodeStack.pop();//访问完毕需要出栈
if (top->right != nullptr) { //当该节点的右节点不为空时,对右节点进行循环访问
cur = top->right;
}
}
return res;
}
};
对于二叉搜索树,我们可以通过中序遍历得到一个递增的有序序列。
后序遍历
后序遍历是先遍历左子树,然后遍历右子树,最后访问树的根节点。
当你删除树中的节点时,删除过程将按照后序遍历的顺序进行。 也就是说,当你删除一个节点时,你将首先删除它的左节点和它的右边的节点,然后再删除节点本身。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
/* 后续遍历的递归方法 */
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
if(root){
process(root,res);
}
return res;
}
void process(TreeNode* root,vector<int>& res){
if(root->left) process(root->left,res);
if(root->right) process(root->right,res);
res.push_back(root->val);
}
};
/* 后续遍历的迭代解法 */
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> nodeStack;
TreeNode* cur=root;
TreeNode* lastnode=nullptr;
while(cur||!nodeStack.empty()){
while(cur){
nodeStack.push(cur);
cur=cur->left;
}
TreeNode* top=nodeStack.top();
if(top->right==nullptr||lastnode==top->right){
res.push_back(top->val);
lastnode=top;
nodeStack.pop();
}else{
cur=top->right;
}
}
return res;
}
};
后序在数学表达中被广泛使用。 编写程序来解析后缀表示法更为容易。
可以使用中序遍历轻松找出原始表达式。 但是程序处理这个表达式时并不容易,因为必须检查操作的优先级。如果想对这棵树进行后序遍历,使用栈来处理表达式会变得更加容易。 每遇到一个操作符,就可以从栈中弹出栈顶的两个元素,计算并将结果返回到栈中。
层序遍历
层序遍历就是逐层遍历树结构。广度优先搜索是一种广泛运用在树或图这类数据结构中,遍历或搜索的算法。 该算法从一个根节点开始,首先访问节点本身。 然后遍历它的相邻节点,其次遍历它的二级邻节点、三级邻节点,以此类推。当我们在树中进行广度优先搜索时,我们访问的节点的顺序是按照层序遍历顺序的。
通常使用队列的数据结构来帮助我们做广度优先搜索。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
/* 递归 有点复杂了 */
class Solution {
public:
vector<vector<int>> res;
vector<vector<int>> levelOrder(TreeNode* root) {
process(root,0);
return res;
}
void process(TreeNode* root,int level){
if(root==NULL) return;
if(res.size()<level+1){
vector<int> tmp{root->val};
res.push_back(tmp);
}else{
res[level].push_back(root->val);
}
process(root->left,level+1);
process(root->right,level+1);
}
};