🏆一、从上到下打印二叉树Ⅰ
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
这道题目考察的是层序遍历,也就是BFS广度优先。不能使用递归求解,因为递归会深度优先。比较难的地方在于我们如何从根的左子树遍历到右子树之后再回到根的左子树的左子树。也就是
遍历到20的时候如何回到9的左子树去遍历11。这是一个难点。
🖊BFS模拟队列
如果我们访问到3这个节点的时候,除了把它的值存下来,还可以用一个容器把9和20这两个节点存下来,依次再去访问9这个节点时,如果9有子节点,就把9的左右子节点11和12存到20这个节点后面,然后访问20这个节点时,把20的子节点15,7存在11,12的后面,这样就可以实现广度优先遍历。
存的前提是pos指向的节点有左子树或者右子树,如果没有就不存,当pos和tail指针重合的时候,代表层序遍历结束。
//BFS广度优先遍历 //模拟一个队列 const int MAX_Node=1001; int* levelOrder(struct TreeNode* root, int* returnSize) { *returnSize=0; if(root==NULL) return NULL; int* ans=(int*)calloc(MAX_Node,sizeof(int)); struct TreeNode* Queue[MAX_Node]; memset(Queue,0,sizeof(struct TreeNode*)); int pos=0; int tail=0; Queue[tail++]=root; while(pos<tail) { ans[(*returnSize)++]=Queue[pos]->val; struct TreeNode* tmp=Queue[pos++]; if(tmp->left) Queue[tail++]=tmp->left; if(tmp->right) Queue[tail++]=tmp->right; } return ans; }
这里用指针数组模拟队列,每一个元素都是节点地址,这种设计很赞。因为题目给我们说了最多不超过1000个节点,针对这一点我们可以设全局变量来开辟空间。
🏆二、从上到下打印二叉树Ⅱ
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
🖊BFS模拟队列返回二维数组
这道题目和上一道本质别无二样,只不过我们需要把它存到二维数组返回。而且二维数组的每一个元素是一个数组,每一个数组里面是这一层的所有结点存储的值,意味着我们要记录每一层结点的个数,每次存储一层,这是和上一道题目不一样的地方。而我们在访问到每一层的结点时需要把它的子节点存到后面。
const int MAX_Node=2000; int** levelOrder(struct TreeNode* root, int* returnSize, int** returnColumnSizes) { *returnSize=0; if(root==NULL) return NULL; int**ans=(int**)calloc(MAX_Node,sizeof(int*)); struct TreeNode* Queue[MAX_Node]; memset(Queue,0,sizeof(struct TreeNode*)); int head,tail=0; Queue[tail++]=root;//把根存下来 *returnColumnSizes=(int*)calloc(MAX_Node,sizeof(int)); while(head<tail) { int size=tail-head; ans[(*returnSize)]=(int*)calloc(size,sizeof(int)); (*returnColumnSizes)[*returnSize]=size; int i=0; while(i<size) { struct TreeNode* tmp=Queue[head++]; ans[(*returnSize)][i++]=tmp->val; if(tmp->left) Queue[tail++]=tmp->left; if(tmp->right) Queue[tail++]=tmp->right; } (*returnSize)++; } return ans; }