Leetcode 102. Binary Tree Level Order Traversal(二叉树的层序遍历)
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example: Given binary tree [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
return its level order traversal as:
[ [3], [9,20], [15,7] ]
分析:二叉树的层序遍历,使用广度优先搜索,建立一个队列q。
每次将队首temp结点出队列的同时,将temp的左孩子和右孩子入队
用len标记入队后的队列长度
row数组始终为当前层的遍历结果,然后用while(len–)语句保证每一次保存入row的只有一层。
每一次一层遍历完成后,将该row的结果存入二维数组v。同时将row中的元素清空。
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10 class Solution { 11 public: 12 vector<vector<int>> levelOrder(TreeNode* root) { 13 vector<int> row; 14 vector<vector<int>> v; 15 if(root == NULL) 16 return v; 17 queue<TreeNode*> node_que;//定义一个队列,元素为节点 18 TreeNode* temp;//每次都指向队列的第一个元素 19 node_que.push(root);//将根节点放入队列 20 while(!node_que.empty()){ 21 int len = node_que.size();//计算当前队列中的元素个数 22 //每次只下一层的节点放入队列中,且每次只将本层的元素放入row 23 while(len--){ 24 temp = node_que.front(); 25 node_que.pop();//弹出队列的第一个元素; 26 row.push_back(temp->val); 27 if(temp->left != NULL)//保证入队列的不会是空节点 28 node_que.push(temp->left); 29 if(temp->right != NULL)//保证入队列的不会是空节点 30 node_que.push(temp->right); 31 } 32 v.push_back(row); 33 row.clear(); 34 } 35 return v; 36 } 37 };
越努力,越幸运