🎁个人主页:我们的五年
🔍系列专栏:初阶初阶结构刷题
🎉欢迎大家点赞👍评论📝收藏⭐文章
目录
1.题目描述:编辑
2.问题分析:
🍔函数解读:
🍔确定数组的大小:
🍔调用遍历函数:
3.最终代码:
前言:
二叉树的遍历顺序有:1.前序:根->左子树->右子树。
2.中序:左子树->根->右子树。
3.后序:左子树->右子->树。
4.层序:一层一层的遍历。
这里我们讲二叉树的前序遍历。力扣题目链接:. - 力扣(LeetCode)
1.题目描述:
2.问题分析:
🍔函数解读:
力扣官方给的函数接口:
int* preorderTraversal(struct TreeNode* root, int* returnSize) {
}
returnSize是数组的长度的地址,不是数组首地址。
而且前面还有这样一句话: * Note: The returned array must be malloced, assume caller calls free().
也就是说,还要malloc动态在堆区上申请一个数组,这样preorderTraversal这个函数销毁时,数组不随函数一起销毁,因为数组在堆区上。而且我们也不要去free(),由caller去free。
🍔确定数组的大小:
我们要去申请数组,那就要确定数组的大小,那么我们把数组的大小顶为多大呢?
题中说了节点数目在[0,100]。
📷给出下面几种情况进行选择:(看看哪种情况最好)
1.因为题目中给了最多为100个节点,所以申请100*sizeof(int)的大小?
2.先申请小一点,不够的话就再去扩容?
3.先去计算树的大小,再去扩容?
分析:
1.如果题目给的是一亿个,我们不可能去申请一亿大小的空间。而且这种情况会有空间浪费。
2.如果频繁的扩容会造成速度很慢,特别是异地扩容,realloc内部自己还要去移动数据。
3.情况三不会浪费空间,又不会频繁扩容。
所以我们先去计算树的节点个数:
分治思想:每次都把树分成三个部分,根+左子树+右子树
最小子问题根为NULL就返回0.
int TreeSize(struct TreeNode* root)
{
if(root==NULL)
return 0;
//分治,总个数等于根+加左子树的个数+右子树的个数
return TreeSize(root->left)+TreeSize(root->right)+1;
}
🍔调用遍历函数:
//i的作用是确定调用该函数的时候,在哪个位置插入。
并且传指针,(*i)++才能改变外部i的值。
如果是传值,i的值不能改变,同一个函数里面左子树i和右子树一样。下面右具体分析:
void preorder(struct TreeNode* root,int* a,int* i)
{
if(root==NULL)
return;
a[(*i)++]=root->val;
preorder(root->left,a,i);
preorder(root->right,a,i);
}
int* preorderTraversal(struct TreeNode* root, int* returnSize) {
*returnSize=TreeSize(root);
int *a=(int*)malloc(sizeof(int)*(*returnSize));
int i=0;
preorder(root,a,&i);
return a;
}
测试用例过了一半,在哪个测试用例就过不了呢?
运行测试用例都能过
如果细心一点的就可以发现,左右子树都有的时候,就过不了,只有一边有树,或者只有根就可以过。
因为这样左右子树开始时都是一样的,如果这样,调用右边的时候,又把左边已经覆盖的值又去覆盖了一遍,所以左边子树的值就没了。
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 TreeSize(struct TreeNode* root)
{
if(root==NULL)
return 0;
return TreeSize(root->left)+TreeSize(root->right)+1;
}
void preorder(struct TreeNode* root,int* a,int i)
{
if(root==NULL)
return;
a[i++]=root->val;
preorder(root->left,a,i);
preorder(root->right,a,i);
}
int* preorderTraversal(struct TreeNode* root, int* returnSize) {
*returnSize=TreeSize(root);
int *a=(int*)malloc(sizeof(int)*(*returnSize));
int i=0;
preorder(root,a,i);
return a;
}