題目描述

輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果，請(qǐng)重建出該二叉樹。假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果中都不含重復(fù)的數(shù)字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6}，則重建二叉樹并返回。

解題思路

用前序的順序重建整個(gè)二叉樹，先判斷傳進(jìn)來參數(shù)是否合法，創(chuàng)建根結(jié)點(diǎn)，然后找到第一個(gè)根結(jié)點(diǎn)位置，遍歷整個(gè)中序順序，找到與前序序列第一個(gè)數(shù)相等的數(shù)的下標(biāo)。然后創(chuàng)建左右子樹，左右子樹里又有左子樹和右子樹。

完整代碼

/*** Definition for binary tree* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/ //首先找出根結(jié)點(diǎn)，先把樹分成左和右，需要準(zhǔn)備兩個(gè)數(shù)組，存放左子樹，和右子樹 class Solution { public:TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {//先判斷合不合法int presize = pre.size();int vinsize = vin.size();if(pre.empty() || vin.empty())return NULL;//因?yàn)橐?#xff0c;所以要?jiǎng)?chuàng)建根結(jié)點(diǎn)TreeNode* root = new TreeNode(pre[0]);//找到中序遍歷的結(jié)點(diǎn)int firstroot = -1;for(int i = 0; i< vinsize; ++i){if(pre[0] == vin[i]){firstroot = i;break;}}//如果沒找到則返回if(firstroot == -1)return NULL;//歸并加遞歸創(chuàng)建左子樹和右子樹vector<int>left_pre,right_pre,left_vin,right_vin;for(int i =0 ;i< firstroot;++i){//在根節(jié)點(diǎn)左邊都是左子樹//左子樹的前序left_pre.push_back(pre[i+1]);//左子樹的中序left_vin.push_back(vin[i]);}for(int i = firstroot+1;i<vinsize;++i){//在根節(jié)點(diǎn)右邊都是右子樹//右子樹的前序right_pre.push_back(pre[i]);//右子樹的中序right_vin.push_back(vin[i]);}//根左右的左root->left = reConstructBinaryTree(left_pre,left_vin);//根左右的右root->right = reConstructBinaryTree(right_pre,right_vin);return root;} };

總結(jié)

以上是生活随笔為你收集整理的【剑指offer】_04 重建二叉树的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。