给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
递归法:其实就是三行代码
inorderTraversal(root->left);
.push_back(root->val);
inorderTraversal(root->right);
/**
* 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) {
//注意子函数调用变量是临时性的,需要将其返回到上一层,与cout 输出显示不同。
vector<int>my_vector;
if(root!=NULL)
{
vector<int> tmp1=inorderTraversal(root->left);
if(!tmp1.empty())
{
for(int i=0;i<tmp1.size();i++)
{
my_vector.push_back(tmp1[i]);
}
}
my_vector.push_back(root->val);
vector<int> tmp2=inorderTraversal(root->right);
if(!tmp2.empty())
{
for(int i=0;i<tmp2.size();i++)
{
my_vector.push_back(tmp2[i]);
}
}
}
return my_vector;
}
};
迭代法
跟着走了一遍过程,怎样想才好理解(在脑子里动态演示一遍过程)
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> v;
stack<TreeNode*> s;
TreeNode* cur = root;
while (cur || !s.empty()) {
if (cur) {
//当前节点压栈
s.push(cur);
//先走左边
cur = cur->left;
} else {
cur = s.top()->right;
//左边走完后记录根节点的值
v.push_back(s.top()->val);
s.pop();
}
}
return v;
}
};