/**
*
* 算法思想:
* 递归的思想,传递变量level来表示处在第几层,使用rcs数组存放每层中已经存放了几
* 个元素。
* 采用 左子树 ->本节点 ->右子树 的遍历顺序,达到从左到右的添加顺序。
* 或者 本节点 -> 左子树 ->右子树 顺序,左边在右边前面即可。
*
*/
#define LEV 1024
#define LEN 1024
typedef struct TreeNode Node;
void get(Node *node, int level, int **ret, int *ret_index, int *rcs){
if(!node) return;
if(!ret[level]){
ret[level] = (int *)malloc(sizeof(int) * LEN);
rcs[level] = 0;
}
if(level > *ret_index){
*ret_index = level;
}
ret[level][rcs[level]++] = node->val;
get(node->left, level+1, ret, ret_index, rcs);
get(node->right, level+1, ret, ret_index, rcs);
}
int** levelOrder(struct TreeNode* root, int* returnSize, int** returnColumnSizes){
int **ret = (int **)malloc(sizeof(int*) * LEV);
int *rcs = (int*)malloc(sizeof(int) * LEV);
int ret_index = 0;
memset(ret, 0, sizeof(int*) * LEV);
memset(rcs, 0, sizeof(int) * LEV);
get(root, 0, ret, &ret_index, rcs);
*returnSize = root==NULL ? 0 : ret_index+1;
*returnColumnSizes = rcs;
return ret;
}
LeetCode-102-二叉树的层次遍历-C语言
最新推荐文章于 2024-07-28 17:42:24 发布