前序,中序遍历序列重建二叉树
输出二叉树的右视图
http://www.nowcoder.com/questionTerminal/c9480213597e45f4807880c763ddd5f0
/* 根据先序遍历和中序遍历重建二叉树步骤: 1.先序的第一个元素pre[pstart]就是根节点,在中序中找到这个根节点的下标k。下标从0开始 2.找到根节点下标后,那么左子树节点个数是:k-istart 3.根节点的左子树寻找范围从pstart+1就可以了,中序的范围是 istart ---- k-1 4.根节点的右子树范围从先序的 pstart + k-istart +1。pos代表在先序序列中右子树根节点的下标。 中序序列中,从k+1到iend即可 */
class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * 求二叉树的右视图 * @param xianxu int整型vector 先序遍历 * @param zhongxu int整型vector 中序遍历 * @return int整型vector */ /* 根据先序遍历和中序遍历重建二叉树步骤: 1.先序的第一个元素pre[pstart]就是根节点,在中序中找到这个根节点的下标k。下标从0开始 2.找到根节点下标后,那么左子树节点个数是:k-istart 3.根节点的左子树寻找范围从pstart+1就可以了,中序的范围是 istart ---- k-1 4.根节点的右子树范围从先序的 pstart + k-istart +1。pos代表在先序序列中右子树根节点的下标。 中序序列中,从k+1到iend即可 */ TreeNode * fun(vector<int> &pre,int pstart,vector<int> &inorder,int istart,int iend) { if(istart>iend) { return NULL; } TreeNode *root=new TreeNode(pre[pstart]); int k=0; for(int i=istart;i<=iend;i++) { if(inorder[i]==root->val) { k=i;//在中序遍历中,下标k是 根节点,比如k=3,istart=0,3-0=3,代表左子树有3个元素 break;//pstart + k-istart + 1。 pstart代表根节点下标,k-istart代表左子树元素个数,+1表示右子树根节点下标 } } root->left=fun(pre,pstart+1,inorder,istart,k-1); root->right=fun(pre,pstart+k-istart+1,inorder,k+1,iend);//pstart+k-istart+1在前序基础上 + 中序上的偏移绝对值 return root; } vector<int> solve(vector<int>& xianxu, vector<int>& zhongxu) { // write code here int n = zhongxu.size(); TreeNode* root = fun(xianxu, 0, zhongxu, 0, zhongxu.size()-1); vector<int> res; if(!root) return res; queue<TreeNode*> que; que.push(root); while(que.size()) { int n=que.size(); for(int i=0;i<n;i++) { TreeNode* node=que.front(); que.pop(); if(i==n-1) { res.push_back(node->val); } if(node->left) { que.push(node->left); } if(node->right) { que.push(node->right); } } } return res; } };