题目:
不分行从上到下打印二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
思路:
乍一看这个题其实就是考的二叉树的遍历,但是属于特殊的层序遍历。对于没有接触过层序遍历的人来说最简单的想法肯定是设计两个函数,一个打印函数一个是遍历函数,先打印根节点,然后打印根节点的左子节点,再打印根结点的右子节点,然后root=root->left;但这个遍历有一个问题就是怎么在root=root->left后再root=root->right,再设置一个变量来储存根节点?显然不显示,对于上层的根节点还好解释,往下层走情况就会变得很复杂了。显然这个思路是行不通的。
所以我们需要用别的媒介来改变这一遍历方式。比如创建一个容器来存储这些节点值。接下来我们大概梳理一下这个思路过程
1.按照题目的意思我们还是按层打印,先打印根节点,为了接下来能够打印根节点的两个子节点,我们遍历到根节点时,我们就把那两个子节点存入,按照从左到右的存储顺序,然后再从容器里取出来,也是从左到右的取出来。
2.所以我们这个容器很显然易见,先进先出,就是队列。打印出根节点的左子节点后就再把左子节点的两个子节点再存入队列中,后面的思路就是基本一致。
代码实现:
/**
* 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().
*/
struct Quene {
struct TreeNode stackin[1000];
struct TreeNode stackout[1000];
struct TreeNode* top_in;
struct TreeNode* top_out;
};
struct Quene* CreatQuene()
{
struct Quene* ret = (struct Quene*)malloc(sizeof(struct Quene));
ret->top_in = ret->stackin;
ret->top_out = ret->stackout;
return ret;
}
void Push(struct TreeNode* root, struct Quene* quene)
{
*(quene->top_in) = *root;
(quene->top_in)++;
}
struct TreeNode Pop(struct Quene* quene)
{
if (quene->top_out == quene->stackout) {
if (quene->top_in == quene->stackin) {
} else {
while (quene->top_in != quene->stackin) {
(quene->top_in)--;
*(quene->top_out) = *(quene->top_in);
(quene->top_out)++;
}
}
}
(quene->top_out)--;
return *(quene->top_out);
}
bool IsEmpty(struct Quene* quene)
{
if ((quene->stackin == quene->top_in) && (quene->stackout == quene->top_out)) {
return true;
}
return false;
}
int* levelOrder(struct TreeNode* root, int* returnSize)
{
if (root == NULL) {
*returnSize = 0;
return NULL;
}
struct Quene* quene = CreatQuene();
Push(root, quene);
int buf[10000];
int q = 0;
while (!IsEmpty(quene)) {
struct TreeNode temp = Pop(quene);
buf[q++] = temp.val;
if (temp.left != NULL) {
Push(temp.left, quene);
}
if (temp.right != NULL) {
Push(temp.right, quene);
}
}
*returnSize = q;
int* ret = (int*)malloc(q * sizeof(int));
memcpy(ret, buf, q * sizeof(int));
return ret;
}
模拟队列:
int* levelOrder(struct TreeNode* root, int* returnSize){
if(!root){
*returnSize=0;
return NULL;
}
*returnSize=0;
int* res=(int*)malloc(sizeof(int)*10000);
struct TreeNode** queue=(struct TreeNode**)malloc(sizeof(struct TreeNode*)*10000);
int head=0,rear=0;
queue[rear++]=root;
while(head<rear){
struct TreeNode* tmp=queue[head++];
res[(*returnSize)++]=tmp->val;
if(tmp->left){
queue[rear++]=tmp->left;
}
if(tmp->right){
queue[rear++]=tmp->right;
}
}
return res;
}