记录每天leetcode刷题
时间:2020年10月19日
问题描述:给定一个二叉树,返回它的中序 遍历。
示例:
了解:二叉树的中序遍历的访问顺序为先访问左孩子结点,再访问根节点,最后访问右孩子结点。
方法:
(1)递归
递归的思路很简单。
void inOrder(TreeNode *root,vector<int>&ans){
if(root== nullptr)
return;
inOrder(root->left,ans);
ans.emplace_back(root->val);
inOrder(root->right,ans);
}
vector<int> inorderTraversal(TreeNode *root) {
vector<int> ans;
inOrder(root,ans);
return ans;
}
(2)迭代
利用栈存储二叉树的结点,不断的遍历左子树,每遍历一次,就将当前结点存储到栈中。当左子树遍历完后,就弹出栈顶元素,记录当前栈顶元素的值,并访问当前结点的右子树,
vector<int> inorderTraversal(TreeNode *root) {
vector<int> res;
stack<TreeNode *> s; //利用栈存储二叉树结点,实现中序遍历
while (root != nullptr || !s.empty()) {
//不断遍历左子树,每遍历一次,就将当前结点存储到栈中
if(root!= nullptr){
s.push(root);
root=root->left;
}else{
//左子树遍历完,则弹出栈顶结点,并访问右子树
auto cur=s.top();
s.pop();
res.emplace_back(cur->val); //存储节点的值
root=cur->right; //访问右子树
}
}
return res;
}
收获:
(1)、中序遍历的迭代实现与前序遍历和后序遍历的迭代的实现不同,需要牢记。