【LeetCode】Easy | 144. 二叉树的前序遍历

Q

思路

如何对树进行前遍历而且将每个数存入数组中？
① 如何遍历：写子程序_preorderTraversal(struct TreeNode*root)内容为递归（根左右）
② 如何存入数组：在子程序中传入数组_preorderTraversal(struct TreeNode*root,int *a)
③ 如何控制数组的下标：
如果传入_preorderTraversal(struct TreeNode*root,int *a，int i)由于是递归形式可能会出现i=2（随便举例子也可能是i=3…）两次（次数随每层节点个数决定），那么就会出现数组a[2]被覆盖的问题。最好的解决方案是传递i的地址，_preorderTraversal(struct TreeNode*root,int *a,int *pi)

代码

/**
 * 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 TreeSzie(struct TreeNode* root)//计算树的节点个数
{
    if(root==NULL) return 0;
    return TreeSzie(root->left)+TreeSzie(root->right)+1;
}
 void _preorderTraversal(struct TreeNode*root,int *a,int *pi)//子程序的命名一般是主程序名字前加_
 {
     if(root==NULL) return;
     printf("%d ",root->val);
     a[(*pi)++]=root->val;
     _preorderTraversal(root->left,a,pi);
     _preorderTraversal(root->right,a,pi);
 }
int* preorderTraversal(struct TreeNode* root, int* returnSize){
    int size=TreeSzie(root);
    int *a=malloc(sizeof(int)*size);//为a开辟空间
    int i=0;
    _preorderTraversal(root,a,&i);
    *returnSize=size;//一定要加这句否则编译不能通过
    return a;
}