前序遍历:中左右
后序遍历:左右中
中序遍历:左中右
144. 二叉树的前序遍历
思路:1.递归遍历,构造递归函数,按照中左右的顺序遍历二叉树的所有元素放入数组中。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
//递归函数
void traversal(TreeNode* cur,vector<int>& result)
{
if(cur==NULL)
{
return;
}
result.push_back(cur->val);
traversal(cur->left,result);
traversal(cur->right,result);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root,result);
return result;
}
};
2、非递归遍历:利用迭代遍历、利用栈先入后出的特点,遍历二叉树的元素放入栈中,先放入根节点,然后弹出栈的头元素(根节点),然后将根节点的右节点和左节点输入栈中,这样的话栈的头部就是左节点,弹出的时候先弹出左节点,只要栈不为空就不结束循环,下个循环中头节点为左节点,此时再弹出它,在判断它的左右节点。这样就能保证树的所有元素都是按前序(左中右)顺序进入的数组。
class Solution {
public:
//利用迭代遍历 栈
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if (root==NULL)
{
return result;
}
else
{
st.push(root);
}
while(!st.empty())
{
TreeNode* node = st.top();
st.pop();
result.push_back(node->val);
if(node->right!=NULL)
{
st.push(node->right);
}
if(node->left!=NULL)
{
st.push(node->left);
}
}
return result;
}
};
145.二叉树的后序遍历
思路:1、递归遍历,跟前序遍历的原理相同,只需要在递归函数中改变顺序。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void traversal(TreeNode* cur,vector<int>& result)
{
if(cur==NULL)
{
return;
}
traversal(cur->left,result);
traversal(cur->right,result);
result.push_back(cur->val);
}
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root,result);
return result;
}
};
2、非递归遍历:迭代遍历:原理同前序遍历,后序遍历的顺序为左右中,因此我们将前序遍历的循环中,左右节点入栈的顺序变化,此时前序的顺序变为中右左, 循环结束后,再将数组翻转,即为左右中顺序。
class Solution {
public:
//迭代遍历 栈
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if(root==NULL)
{
return result;
}
else
{
st.push(root);
}
while(!st.empty())
{
TreeNode*node = st.top();
st.pop();
result.push_back(node->val);
if(node->left!=NULL)
{
st.push(node->left);
}
if(node->right!=NULL)
{
st.push(node->right);
}
}
reverse(result.begin(),result.end());
return result;
}
};
94.二叉树的中序遍历
思路:1、递归遍历:原理同前序、后序,秩序将递归函数中的顺序改为左中右。
class Solution {
public:
void traversal(TreeNode* cur,vector<int>& result)
{
if(cur==NULL)
{
return;
}
traversal(cur->left,result);
result.push_back(cur->val);
traversal(cur->right,result);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root,result);
return result;
}
};
2、非递归遍历:迭代遍历,先将头节点放入栈中,然后定义一个指针指向根节点,只要根节点不为空,或者栈不为空,循环就不停止,循环内部,如果指针不为空,就把指针处的节点放入栈中,然后指针指向节点左下节点,找到树的最左节点。当指针指向空时,说明上一个节点没有左节点,将指针指向上一个节点,即栈顶,然后把栈顶元素弹出,把这个元素的值输入数组,然后再判断这个节点的右下节点,如果没有右下节点,继续将指针指向栈顶元素,重复上述循环。如果有右下节点,将这个节点放入栈中,找这个节点是否有左下节点,重复循环。
class Solution {
public:
//迭代遍历
vector<int> inorderTraversal(TreeNode* root) {
//定义一个数组用于记录结果 一个栈用来记录遍历的元素
stack<TreeNode*> st;
TreeNode* cur = root;
vector<int> result;
while(cur!=NULL||!st.empty())
{
if(cur!=NULL)
{
st.push(cur);
cur=cur->left;
}
else
{
cur = st.top();
st.pop();
result.push_back(cur->val);
cur =cur->right;
}
}
return result;
}
};
收获:
1、开始补春节欠的债,第一天接触二叉树,学会了二叉树的基本概念。
2、递归遍历中只需要改动递归函数中的顺序,即可实现前序、后序、中序的遍历。