力扣细节题:二叉树前序遍历

力扣细节题:二叉树前序遍历_第1张图片

细节一:题目在note部分提示我们要使用malloc函数去创建数组,因为它没有提供给我们数组,而题目中的指针returnsize提醒我们这是一个变量进行传址操作用于释放之前malloc的空间,粗暴地按题目范围malloc空间会报错,所以必须使用函数事先测出二叉树的节点数,然后进行递归操作赋值数组即可(注意i++与++i),但此时又报错,而报错原因为当我们进行多次递归操作时,每个函数栈帧都有自己的i导致了逻辑混乱,所以我们必须将i的地址传进去。而由于oj题的特性,在每次使用此函数前都要将i赋值为0

void good(struct TreeNode*root,int*a,int *pi)
 {
if(root==NULL)
{
    return;
}
a[(*pi)++]=root->val;

good(root->left,a,pi);
good(root->right,a,pi);
 }
 int treesize(struct TreeNode*root)
 {
     if(root==NULL)
     {
         return 0;
     }

     return treesize(root->left)+treesize(root->right)+1;
 }
int* preorderTraversal(struct TreeNode* root, int* returnSize) {
    int*a=(int*)malloc(treesize(root)*sizeof(int));
    *returnSize=treesize(root);
    int i=0;
    good(root,a,&i);
    return a;
}

你可能感兴趣的:(数据结构,leetcode,算法,职场和发展)