【C++牛客网刷题】-重建二叉树!
Task
思路:
解这个题之前我是根本不知道树的, 所以必须先要了解什么事前序遍历和中序遍历: 前序遍历:有左走左,无左走右,都无后退 中序遍历:只认准左孩子的左孩子…万不得已才后退,后退后也是上一个节点的右孩子,这样类推! 找了个说法:讲的还算透彻!
有了这个基础字后,就可以重建二叉树了! 后面的还是需要对树有一个深入的理解,要结合剑指Offer来思考和复习!这样 快速一些!
代码
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) { int vinlen = vin.size(); if(vinlen==0) return NULL; vector<int> left_pre, right_pre, left_vin, right_vin; //创建跟姐蒂娜,跟积淀肯定是前序遍历的第一个数 TreeNode* head = new TreeNode(pre[0]); //找到中序遍历根节点所在位置,存放于变量gen中 int gen = 0; for( int i=0; i<vinlen; i++) { if(vin[i] == pre[0]) { gen = i; break; } } //对于中序遍历,根节点左边的节点位于二叉树的左边,根节点右边的节点位于二叉树的右边 //利用上述这点,对二叉树归并 for(int i=0; i<gen; i++) { left_vin.push_back(vin[i]); left_pre.push_back(pre[i+1]); //前序第一个为根节点 } for(int i=gen+1; i<vinlen; i++) { right_vin.push_back(vin[i]); right_pre.push_back(pre[i]); } //和shell排序的思想类似,去除前序和中序遍历根节点左边和右边的子树 //递归,再对其进行上述所有步骤,既再区分子树的左右子树,直到叶节点 head->left=reConstructBinaryTree(left_pre, left_vin); head->right=reConstructBinaryTree(right_pre, right_vin); return head; } };