- 递归
/**
* 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) {
if(root!=nullptr)
{
inorderTraversal(root->left);
res.push_back(root->val);
inorderTraversal(root->right);
}
return res;
}
private:
vector<int> res;
};
2.非递归,会改变树
使用一个栈来存储二叉树节点,根据中序遍历的规则,我们可以推算出这样的规律:
- 将当前非空节点入栈
- 如果左子节点不为空,则继续将左子节点入栈
- 如果左子节点为空,则抛出栈顶节点并记录 val 值,然后将其右子节点入栈
- 重复 1、2、3 步骤直至栈空
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> tmp;
vector<int> res;
if(root!=nullptr) tmp.push(root);
while(!tmp.empty())
{
TreeNode* cur_node = tmp.top();
if(cur_node->left) //左孩子不空,入栈
{
tmp.push(cur_node->left);
cur_node->left=nullptr; //左孩子入栈后置为空指针,否则会死循环(因为这个节点的左孩子已经入栈了,另其为空,避免在弹栈时将左孩子在入栈,陷入死循环)
}
else //左孩子为空,弹栈,入右孩子
{
res.push_back(cur_node->val);
tmp.pop();
if(cur_node->right)
tmp.push(cur_node->right);
}
}
return res;
}
};
- 不改变树
对于一个节点来说,不为空,压栈,指左
为空,弹栈,指右
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> tmp;
vector<int> res;
TreeNode *cur = root;
while(!tmp.empty()||cur) //栈不为空或者指针不为空都继续
{
if(cur) //当前节点不为空则先入栈
{
tmp.push(cur);
cur = cur->left;
}
else //节点为空,输出栈顶元素,栈内所有元素的左孩子都是已经入栈的,故处理右孩子
{
cur = tmp.top();
tmp.pop();
res.push_back(cur->val);
cur = cur->right;
}
}
return res;
}
};