1. 程式人生 > >算法之重建二叉樹

算法之重建二叉樹

算法 code class bsp ons 重建 value col color

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重復的數字。
例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹並返回。

 //m,n 前序數組的起點和終點 i,j 中序數組的起點和終點
TreeNode * ConstructSub(vector<int>&preOrderVec,vector<int>&inOrderVec,int m,int n,int i,int j){
    int rootValue = preOrderVec[m];
    TreeNode 
* root = new TreeNode(rootValue); root->left=nullptr; root->right=nullptr; //邊界, 左邊或者右邊只有一個元素的時候,並且前序和中序的值相等 if(m==n && i==j){ if(preOrderVec[m]==inOrderVec[i]){ return root; } else{return nullptr;} } //找到左右兩邊
//中序序列裏的root的索引 int rootInorderIndex=i; //往後開始找 while (rootInorderIndex<n&&inOrderVec[rootInorderIndex]!=rootValue) { rootInorderIndex++; } //找到了,那就劃分左右子樹,然後遞歸 int leftLength= rootInorderIndex-i; int leftPreorderEndIndex= m+leftLength; //存在左子樹 if
(leftLength>0){ root->left = ConstructSub(preOrderVec, inOrderVec, m+1, leftPreorderEndIndex,i,rootInorderIndex-1); } //存在右子樹 if(leftLength<n-m){ root->right= ConstructSub(preOrderVec, inOrderVec, leftPreorderEndIndex+1, n, rootInorderIndex+1, j); } return root; }

算法之重建二叉樹