Leetcode144. 二叉树的前序遍历(C语言)
数据结构-树:算法与数据结构参考
题目:
给定一个二叉树,返回它的 前序 遍历。(进阶: 递归算法很简单,你可以通过迭代算法完成吗?)例:
输入: [1,null,2,3]
输出: [1,2,3]
思路:
前序遍历(根,左,右),用一个栈存储右节点;迭代遍历,根的顺序即输出结果。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* preorderTraversal(struct TreeNode* root, int* returnSize){
int top=-1;
struct TreeNode** s =(struct TreeNode**)malloc(sizeof(struct TreeNode*)*100); //用来存储遍历所在层的上一层
*returnSize=0;
int *result=(int*)malloc(sizeof(int)*100);
if(!root) return NULL;
s[++top]=root;5
while(top!=-1){
root=s[top--]; //每次均弹出一个(左),原右子树根节点仍然在s中
result[(*returnSize)++]=root->val; //最终结果(根左右)
if(root->right) s[++top]=root->right;
if(root->left) s[++top]=root->left; //存储左右子树根节点
}
return result;
}