二叉树三种遍历方式的六种实现方法 和 层次遍历

基本概念不再赘述，下面直接给出代码：

#include <iostream>
#include <string>
#include <stack>
#include <vector>
using namespace std;
struct Node
{
	int val;
	Node * left;
	Node * right;
	Node(int x) :val(x), left(nullptr), right(nullptr){};
};
 
 
void creatBiTree(Node* &root)
{
	int x;
	cin >> x;
	if (x == -1)
	{
		root = nullptr;
		return;
	}
	root = new Node(x);
	creatBiTree(root->left);
	creatBiTree(root->right);
}
void visit(Node *T)
{
	if (T->val != -1)
		cout << T->val << " ";
 
}


//-----------------------------递归方式遍历------------------------------------
//先序递归遍历
void preOrder(Node* root)
{
	if (root != nullptr)
	{
		visit(root);
		preOrder(root->left);
		preOrder(root->right);
	}
}


//中序递归遍历
void inOrder(Node* root)
{
	if (root != nullptr)
	{
		inOrder(root->left);
		visit(root);
		inOrder(root->right);
	}
}


//后序递归遍历
void postOrder(Node* root)
{
	if (root != nullptr)
	{
		postOrder(root->left);
		postOrder(root->right);
		visit(root);
	}
}


//-----------------------------非递归方式遍历------------------------------------
//先序遍历
void preOrderF(Node* root)
{
	if (root == nullptr)
		return;
	stack<Node *> s;
	s.push(root);
	Node *p = nullptr;
	while (!s.empty())
	{
		p = s.top();
		s.pop();
		cout << p->val << " ";
		if (p->right)  // 注意这里左右子节点的入栈顺序！
			s.push(p->right);
		if (p->left)
			s.push(p->left);
	}
}


//中序遍历
void inOrderF(Node* root)
{
	if (root == nullptr)
		return;
	stack<Node *> s;
	Node *p = root;  // 注意这里并没有提前把root给push进栈（区别于前序和后序的非递归方式遍历）
	while (p || !s.empty())  //这里注意循环条件
	{
		if (p)
		{
			s.push(p);
			p = p->left;
		}
		else 
		{
			p = s.top();
			s.pop();
			cout << p->val << " ";
			p = p->right;
		}
	}
	
}


// 这其实是力扣的一个Hard题，不用技巧还挺麻烦的，记一下这个技巧
// 后序遍历（这是一种非常巧妙的办法，其他办法请参考References）   技巧：LRD-> 逆序输出 -> DRL
// 我们的思路就是：DRL遍历二叉树，最后将得到的结果reverse一下就可以了
void postOrderF(Node* root)
{
	if (root == nullptr)
		return;
	stack<Node *> s;
	vector<int> rs;
	s.push(root);
	Node *p = nullptr;
	while (!s.empty())
	{
		p = s.top();
		s.pop();
		rs.insert(rs.begin(),p->val);  // 其实这里直接push_back()，然后最后reverse一下也可以
		if (p->left)  // 注意这里左右子节点的入栈顺序！   cf: 前序遍历的非递归方式
			s.push(p->left);
		if (p->right)
			s.push(p->right);
	}
	for (int i = 0; i < rs.size(); i++)
		cout << rs[i] << " ";
}


int _tmain(int argc, _TCHAR* argv[]){
	Node * root;
	//二叉树的创建(根据先序创建)
	creatBiTree(root);
	//二叉树的递归遍历
	preOrderF(root);
	cout << endl;
	inOrderF(root);
	cout << endl;
	postOrderF(root);
	cout << endl;
	return 0;
}

以上就是“二叉树三种遍历方式的六种实现方法”，应该比较浅显易懂了，要想了解更多，请参考References。

层次遍历代码如下：

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int> res;
        if(root == nullptr)
            return res;
        
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            TreeNode* node = q.front();
            q.pop();
            res.push_back(node->val);
            
            if(node->left)
                q.push(node->left);
            if(node->right)
                q.push(node->right);
        }
        
        return res;
    }
};

// 进阶形式，逐层打印二叉树（必会！）
class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode* pRoot) {
        vector<vector<int> > res;
        if(pRoot == nullptr)
            return res;
        
        queue<TreeNode*> q;
        q.push(pRoot);
        while(!q.empty()){
            int cnt = q.size();  // 关键：这个size可以保证逐层遍历二叉树
            vector<int> temp;
            while(cnt--){
                TreeNode* pNode = q.front(); q.pop();
                temp.push_back(pNode->val);
                
                if(pNode->left)
                    q.push(pNode->left);
                if(pNode->right)
                    q.push(pNode->right);
            }
            
            res.push_back(temp);
        }
        
        return res;
    }
};

References

• https://blog.csdn.net/u014465639/article/details/71076092/
• https://blog.csdn.net/Mr_HHH/article/details/80212130
• https://blog.csdn.net/dearbaba_8520/article/details/81661663
• https://blog.csdn.net/u013161323/article/details/53925313
• https://blog.csdn.net/zhangxiangDavaid/article/details/37115355