本文實例講述了C++基于先序、中序遍歷結果重建二叉樹的方法。分享給大家供大家參考，具體如下：
題目：
輸入某二叉樹的前序遍歷和中序遍歷的結果，請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重復的數(shù)字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6}，則重建二叉樹并返回。
實現(xiàn)代碼：
#include <iostream>#include <vector>#include <stack>using namespace std;struct TreeNode { int val; TreeNode *left; TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) {}};//創(chuàng)建二叉樹算法TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> mid){ int nodeSize = mid.size(); if (nodeSize == 0) return NULL; vector<int> leftPre, leftMid, rightPre, rightMid; TreeNode* phead = new TreeNode(pre[0]); //第一個當是根節(jié)點 int rootPos = 0; //根節(jié)點在中序遍歷中的位置 for (int i = 0; i < nodeSize; i++) { if (mid[i] == pre[0]) { rootPos = i; break; } } for (int i = 0; i < nodeSize; i++) { if (i < rootPos) { leftMid.push_back(mid[i]); leftPre.push_back(pre[i + 1]); } else if (i > rootPos) { rightMid.push_back(mid[i]); rightPre.push_back(pre[i]); } } phead->left = reConstructBinaryTree(leftPre, leftMid); phead->right = reConstructBinaryTree(rightPre, rightMid); return phead;}//打印后續(xù)遍歷順序void printNodeValue(TreeNode* root){ if (!root){ return; } printNodeValue(root->left); printNodeValue(root->right); cout << root->val<< " ";}int main(){ vector<int> preVec{ 1, 2, 4, 5, 3, 6 }; vector<int> midVec{ 4, 2, 5, 1, 6, 3 }; cout << "先序遍歷序列為 1 2 4 5 3 6" << endl; cout << "中序遍歷序列為 4 2 5 1 6 3" << endl; TreeNode* root = reConstructBinaryTree(preVec, midVec); cout << "后續(xù)遍歷序列為 "; printNodeValue(root); cout << endl; system("pause");}/*測試二叉樹形狀： 1 2 3 4 5 6*/運行結果：
先序遍歷序列為 1 2 4 5 3 6中序遍歷序列為 4 2 5 1 6 3后續(xù)遍歷序列為 4 5 2 6 3 1請按任意鍵繼續(xù). . .
希望本文所述對大家C++程序設計有所幫助。
新聞熱點
疑難解答
