牛客題霸 [輸出二叉樹的右視圖] C++題解/答案

題目描述

請根據二叉樹的前序遍歷，中序遍歷恢復二叉樹，并打印出二叉樹的右視圖

題解；

分兩個過程：
先用前序遍歷+中序遍歷恢復二叉樹，這個應該都會。。
打印二叉樹的后視圖，其實就是層序遍歷中每一層的最后一個元素

代碼；

我的代碼不知為何錯了，，調不出來了。。

class Solution { public:/*** 代碼中的類名、方法名、參數名已經指定，請勿修改，直接返回方法規定的值即可* 求二叉樹的右視圖* @param xianxu int整型vector 先序遍歷* @param zhongxu int整型vector 中序遍歷* @return int整型vector*/vector<int>res;vector<int> solve(vector<int>& xianxu, vector<int>& zhongxu) {// write code hereTreeNode* t=rebulid(xianxu,zhongxu,0,0,zhongxu.size()-1);right(t);return res;}TreeNode* rebulid(vector<int> &pre,vector<int> &in,int root,int l,int r){if(l>r)return NULL;TreeNode* tree=new TreeNode(pre[root]);int mid=l;while(mid<r&&pre[root]!=in[mid])mid++;tree->left=rebulid(pre, in, root+1, l, mid);tree->right=rebulid(pre,in,root+1+mid-l,mid+1,r);return tree;}void right(TreeNode *root){queue<TreeNode*>q;if(!root)return ;q.push(root);while(!q.empty()){int size=q.size();res.push_back(q.front()->val);while(size--){TreeNode* node=q.front();q.pop();if(node->right)q.push(node->right);if(node->left)q.push(node->right);}}} };

借鑒的某大佬代碼：

class Solution { public:/*** 代碼中的類名、方法名、參數名已經指定，請勿修改，直接返回方法規定的值即可* 求二叉樹的右視圖* @param xianxu int整型vector 先序遍歷* @param zhongxu int整型vector 中序遍歷* @return int整型vector*/TreeNode* mytree(vector<int> x,vector<int> z){if(x.size()==0||z.size()==0)return nullptr;TreeNode* root=new TreeNode(x[0]);int t=distance(z.begin(),find(z.begin(),z.end(),x[0]));vector<int> xleft(x.begin()+1,x.begin()+t+1);vector<int> xright(x.begin()+t+1,x.end());vector<int> zleft(z.begin(),z.begin()+t);vector<int> zright(z.begin()+t+1,z.end());root->left=mytree(xleft,zleft);root->right=mytree(xright,zright);return root;}vector<int> solve(vector<int>& xianxu, vector<int>& zhongxu) {// write code hereTreeNode* root=mytree(xianxu,zhongxu);vector<int> res;queue<TreeNode*> que;if(root==nullptr) return res;que.push(root);while(!que.empty()){int len=que.size();for(int i=0;i<len;i++){TreeNode* t=que.front();que.pop();if(i==len-1)res.emplace_back(t->val);if(t->left) que.push(t->left);if(t->right) que.push(t->right);}}return res;} };

總結

以上是生活随笔為你收集整理的牛客题霸 [输出二叉树的右视图] C++题解/答案的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。