目录
一、题目描述
二、初次解答
三、官方解法
四、总结
一、题目描述
二、初次解答
1. 思路:二叉树的中序遍历。访问二叉树的左子树,再访问二叉树的根节点,最后访问二叉树的右叉树。
2. 代码:
void order(struct TreeNode* root, int* ret, int* p){ if(!root) return; order(root->left, ret, p); ret[(*p)++]=root->val; order(root->right, ret, p); } int* inorderTraversal(struct TreeNode* root, int* returnSize) { int* ret=(int*)malloc(sizeof(int)*100); *returnSize=0; order(root, ret, returnSize); return ret; }
3. 优点:实现简单,容易想到,且仅需遍历一遍,时间复杂度为O(n)。
4. 缺点:需要递归栈空间,空间复杂度为O(n)。
三、官方解法
1. 思路:迭代遍历二叉树,手动维护栈。每次迭代访问子节点前,将当前节点地址保存到栈内,访问完左节点后,再访问当前节点,最后访问右节点。
2. 代码:
int* inorderTraversal(struct TreeNode* root, int* returnSize) { int* ret=malloc(sizeof(int)*100); struct TreeNode** tmp=malloc(sizeof(struct TreeNode*)*100); int top=0; *returnSize=0; while(root || top>0){ while(root){ tmp[top++]=root; root=root->left; } root=tmp[--top]; ret[(*returnSize)++]=root->val; root=root->right; } return ret; }
3. 优点:符合不采用迭代的要求,且时间复杂度为O(n)。
4. 缺点:手动维护栈,空间复杂度依旧为O(n)。
四、总结
当遇到二叉树中序遍历时,使用递归实现最简单,也可以采用迭代手动维护栈空间来实现。