原題
題意:
根據(jù)先序和中序得到二叉樹(假設(shè)無重復(fù)數(shù)字)
思路:
先手寫一次轉(zhuǎn)換過程饺蔑,得到思路巍耗。
即從先序中遍歷每個元素,(創(chuàng)建一個全局索引炬太,指向當(dāng)前遍歷到的元素)在中序中找到該元素作為當(dāng)前的root,以該節(jié)點(diǎn)左邊所有元素作為當(dāng)前root的左支亲族,右同理炒考。
重復(fù)分別對左右邊所有元素做相同處理。
class Solution
{
public:
TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
{
int pos = 0;
return buildBinary(preorder, inorder, 0, preorder.size(), pos);
}
private:
TreeNode *buildBinary(vector<int> &preorder, vector<int> &inorder, int beg,
int end, int &pos)
{
TreeNode *node = NULL;
if (beg < end)
{
int i = 0;
for (i = beg; i < end; i++)
{
if (preorder[pos] == inorder[i])
break;
}
++pos;
node = new TreeNode(inorder[i]);
node->left = buildBinary(preorder, inorder, beg, i, pos);
node->right = buildBinary(preorder, inorder, i + 1, end, pos);
}
return node;
}
};