LeetCode 199. 二叉树的右视图
思路
方法:层序遍历
保存层序遍历每一层的最后一个数字即可。
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode() : val(0), left(nullptr), right(nullptr) {} 8 * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} 9 * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} 10 * }; 11 */ 12 class Solution { 13 public: 14 vector<int> rightSideView(TreeNode* root) { 15 vector<int> res; 16 if(root == NULL) 17 return res; 18 19 //层序遍历 20 queue<TreeNode*> q; 21 q.push(root); 22 TreeNode* t; 23 while(!q.empty()) { 24 int qsize = q.size(); 25 for(int i = 0; i < qsize; ++i) { 26 t = q.front(); 27 //保存每一层的最后一个结点的值 28 if(i == qsize-1) { 29 res.push_back(t->val); 30 } 31 if(t->left) q.push(t->left); 32 if(t->right) q.push(t->right); 33 q.pop(); 34 } 35 } 36 37 return res; 38 } 39 };
复杂度分析:
时间复杂度:O(n)
空间复杂度:O(n)