给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
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<vector<int>> rst; 14 vector<int> tmpRst; 15 queue<TreeNode*> q; 16 if(root) q.push(root); 17 int levelcnt=1;//当前层元素个数 18 int count=0; //下一层元素个数 19 while(!q.empty()) 20 { 21 TreeNode* Nd=q.front(); 22 tmpRst.push_back(Nd->val); 23 q.pop(); 24 --levelcnt; 25 if(Nd->left) 26 { 27 q.push(Nd->left); 28 ++count; 29 } 30 if(Nd->right) 31 { 32 q.push(Nd->right); 33 ++count; 34 } 35 if(levelcnt==0) 36 { 37 levelcnt=count; 38 count=0; 39 rst.push_back(tmpRst); 40 tmpRst.clear(); 41 } 42 43 } 44 return rst; 45 } 46 };