C++先序、中序遍歴結果に基づいて二叉木を再構築する方法
2552 ワード
本明細書の例では、C++が、先行シーケンス、中間シーケンスの遍歴結果に基づいてツリーを再構築する方法について説明する.皆さんの参考にしてください.具体的には以下の通りです.
タイトル:
ツリーの前順および中順の結果を入力し、ツリーを再構築します.入力された前シーケンスおよび中シーケンスの結果に重複する数値が含まれていないと仮定します.例えば、前シーケンス遍歴シーケンス{1,2,4,7,3,5,6,8}と中シーケンス遍歴シーケンス{4,7,2,1,5,3,8,6}を入力すると、ツリーが再構築され、戻ります.
実装コード:
実行結果:
本稿で述べたことが皆さんのC++プログラム設計に役立つことを願っています.
タイトル:
ツリーの前順および中順の結果を入力し、ツリーを再構築します.入力された前シーケンスおよび中シーケンスの結果に重複する数値が含まれていないと仮定します.例えば、前シーケンス遍歴シーケンス{1,2,4,7,3,5,6,8}と中シーケンス遍歴シーケンス{4,7,2,1,5,3,8,6}を入力すると、ツリーが再構築され、戻ります.
実装コード:
#include
#include
#include
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
//
TreeNode* reConstructBinaryTree(vector pre, vector mid)
{
int nodeSize = mid.size();
if (nodeSize == 0)
return NULL;
vector leftPre, leftMid, rightPre, rightMid;
TreeNode* phead = new TreeNode(pre[0]); //
int rootPos = 0; //
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;
}
//
void printNodeValue(TreeNode* root)
{
if (!root){
return;
}
printNodeValue(root->left);
printNodeValue(root->right);
cout << root->val<< " ";
}
int main()
{
vector preVec{ 1, 2, 4, 5, 3, 6 };
vector 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 << " ";
printNodeValue(root);
cout << endl;
system("pause");
}
/*
:
1
2 3
4 5 6
*/
実行結果:
1 2 4 5 3 6
4 2 5 1 6 3
4 5 2 6 3 1
. . .
本稿で述べたことが皆さんのC++プログラム設計に役立つことを願っています.