Leetcode144. 二叉树的前序遍历（C语言）

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;
}