1.二叉树的遍历
提到二叉树,遍历是其不可缺省的一部分,在大家已经了解了二叉树前中后序遍历的原理以及方式,接下来选择考虑相关题目来巩固知识。
1.中序遍历
首先是中序遍历,后序遍历的顺序就是左子树,根,右子树。所以我们使用递归的逻辑也就是左->根->右。
先贴一下OJ的代码:
int i = 0;
int TreeSize(struct TreeNode *root){
if(root == NULL){
return 0;
}
return TreeSize(root->left) + TreeSize(root->right) + 1;
}
void inorder(struct TreeNode* root, int *a){
if(root == NULL){
return;
}
inorder(root->left,a);
a[i++] = root->val;
inorder(root->right,a);
}
int* inorderTraversal(struct TreeNode* root, int* returnSize) {
int size = TreeSize(root);
*returnSize = size;
int *a = (int*)malloc(sizeof(int) * size);
i = 0;
inorder(root,a);
return a;
}
首先我们来看一下OJ提供的函数。int* postorderTraversal(struct TreeNode* root, int* returnSize),这个函数的接口参数是当前节点以及返回数组的数据大小(此处数组是用于后序遍历序列存储),返回参数是数组的首地址。
其次,对于数组的大小,不选择用硬编码的格式·,防止浪费空间或开辟空间不够,用TreeSize函数进行对二叉树遍历求其结点个数。这个问题下我们使用递归,即考虑,子问题和条件,当结点为空时,表示此时不存在节点,返回0。如果不空,遍历左右再+1。下面,就是动态开辟内存。
inorder这个函数就是中序遍历,由于不好在原函数进行递归,此处选择外部调用。其实逻辑很简单,就是左子树递归,存储,右子树递归,需要注意的一点就是,此时控制自增变量的时候需要选择全局变量而不是选择局部变量。
此处做一个粗略的图,如果此时不采用全局变量,由于在递归中函数在不同的函数栈帧中,其中的i属于不同值的变量,当左子树递归完成后此时局部变量的i存的值会被右子树的值覆盖而不能进行正常的自增,所以这个时候选用全局变量作为自增变量。只需要在用完后回零即可。
2.前序遍历
基本与上述代码段类似,只需要在遍历函数中做如下修改即可
if(root == NULL){
return;
}
a[i++] = root->val;
inorder(root->left,a);
inorder(root->right,a);
3.后序遍历
同上。