leetcode105
题目简介
通过前序遍历和中序遍历来确定二叉树的形状
(PS:对于新手来说,一开始有点懵,这个树怎么建,后来发现上面有结构体来描述节点,才知道使用节点通过递归相互联结)
题目思想
由于输入是向量形式,于是我们可以通过下标来确定节点的左右节点。这是一种经典的分治算法,首先通过先序遍历确定根节点,然后中序遍历中根节点左右就是他的左右子树,然后依次递归即可。
AC代码
/**
1. Definition for a binary tree node.
2. struct TreeNode {
3. int val;
4. TreeNode *left;
5. TreeNode *right;
6. TreeNode(int x) : val(x), left(NULL), right(NULL) {}
7. };
*/
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int root=0;
return buildtree2(preorder,inorder,root,0,preorder.size()-1);
}
TreeNode* buildtree2(vector<int>& preorder, vector<int>& inorder,int& root, int left, int right){
int i=left;
if(root>=preorder.size()){
return 0;
}
for(i=left;i<=right;i++){
if(inorder[i]==preorder[root]){
break;
}
}
TreeNode* prient= new TreeNode(preorder[root]);
root++;
if(left<=i-1){
prient->left=buildtree2(preorder,inorder,root,left,i-1);
}
if(right>=i+1){
prient->right=buildtree2(preorder,inorder,root,i+1,right);
}
return prient;
}
};
代码小结:
- 对于root这个变量是指向当前根节点,并且root一定要是引用的(&),因为他是全局的变量,而不是函数内的,所以要用引用。
- 分治方法顾名思义是分治,我们使用的输入是向量,所以我们可以使用下标来确定每次所用到的位置