原题链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/
题目:
给定一个二叉树的根节点 root ,返回它的 中序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,3,2]
思路:
中序遍历罢了。
三个遍历的伪代码:
//前序遍历
void PreOrder(TreeNode* head) {
if (head == nullptr) return;
cout<<head->value;
PreOrder(head->left);
PreOrder(head->right);
}
//中序遍历
void MidOrder(TreeNode* head) {
if (head == nullptr) return;
MidOrder(head->left);
cout<<head->value;
MidOrder(head->right);
}
//后序遍历
void PostOrder(TreeNode* head) {
if (head == nullptr) return;
PostOrder(head->left);
PostOrder(head->right);
cout<<head->value;
}
按照中序的思路进行编写,比较简单。
/**
* 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:
vector<int> res = {};
void MidOrder(TreeNode* head) {
if (head == nullptr) {
return;
}
MidOrder(head->left);
//将该值添加到结果数组中
res.push_back(head->val);
MidOrder(head->right);
}
vector<int> inorderTraversal(TreeNode* root) {
if (root == nullptr) {
return {};
}
MidOrder(root);
return res;
}
};
思路二:
迭代。
首先明确一下迭代和递归的区别吧,迭代可以说是由近及远,算f(n)的时候从f(1)开始算起,推到f(2)逐渐到f(n),也可以说是前一步的输出作为后一步的输入。
递归呢就是由远及近,算f(n)的时候先算f(n-1),为了算f(n-1)再算f(n-2)直至f(1)。用图的话可以画成这个样子。
】
】】
】】】
】】】】
】】】】】
】】】】
】】】
】】
】
大概这个意思吧。
函数表示的话
//求阶乘
//迭代,重复一定的算法
int fun(int x) {
int sum = 1;
for (int i = x; i > 0; i--) {
sum = sum * i;
}
return sum;
}
//递归,重复自身的方法
int fun(int x) {
if (x > 0) {
return x * fun(x -1);
} else {
return 1;
}
}
OK,那么迭代咋搞呢。
迭代利用的是栈。
思路就是深度搜索这个树,深搜时入栈,为空和回退时出栈,并且保存该值到结果数组中。最后返回。
/**
* 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:
vector<int> res = {};
void MidOrder(TreeNode* head) {
vector<int> res = {};
stack<TreeNode*> stk;
while(root != nullptr || !stk.empty()) {
while(root != nullptr) {
stk.push(root);
root = root->left;
}
root = stk.top();
res.push_back(root->val);
stk.pop();
root = root->right;
}
return res;
}
};