2016-09-18 65 views
1

我想在C++中用它的前序和ignorer序列来构建一个二叉树(不重复)。为什么离开函数时节点值发生了变化?

编码定义如下:

#include <iostream> 
#include <string> 
#include <vector> 
#include <cmath>  

using namespace std; 

struct TreeNode { 
    int val; 
    TreeNode *left; 
    TreeNode *right; 
    TreeNode(int x): val(x),left(NULL),right(NULL) {} 
}; 

int idx = 0; //idx in preorder 
void helper(TreeNode* node,int start,int end, vector<int>& preorder, vector<int>& inorder) { 
    if (start == end) return; 
    int dummy; 
    for (dummy = 0; dummy<inorder.size(); dummy++) { 
     if (inorder[dummy] == node->val) { 
      break; 
     } 
    } 
    idx ++; 
    TreeNode left = TreeNode(preorder[idx]); 
    node->left = &left; 
    helper(node->left,start,dummy-1,preorder,inorder); 
    idx ++; 
    TreeNode right = TreeNode(preorder[idx]); 
    node->right = &right; 
    helper(node->right,dummy+1,end,preorder,inorder); 
} 

TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { 
    TreeNode root = TreeNode(preorder[0]); 
    helper(&root,0,preorder.size()-1,preorder,inorder); 
    return &root; 
}  

int main(int argc, const char * argv[]) { 
    // insert code here.. 
    // build the tree 
    vector<int> preorder = {2,1,3}; 
    vector<int> inorder = {1, 2, 3}; 

    TreeNode* root = buildTree(preorder,inorder); 
    return 0; 
} 

它什么也没有返回,当我试图调试它一步一步,我发现,当我离开的左子和右孩子的值发生变化功能。我已经重写了代码并使用了新的TreeNode,它可以工作。

但是我仍然对以前的版本感到困惑。这是由我用来创建TreeNode的方式引起的吗?

我不熟悉C++指针,所以任何提示都非常感谢!

回答

3
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { 
    TreeNode root = TreeNode(preorder[0]); 
    helper(&root,0,preorder.size()-1,preorder,inorder); 
    return &root; 
} 

这里您返回本地创建的对象的地址(它将指向函数返回后的某个随机内存位置)。您需要使用new在堆上创建根目录。例如:

TreeNode* root = new TreeNode(preorder[0]); 
//stuff... 
return root; 

您的代码中存在更多的错误......但修复上面的指针将是帮助您调试它们的第一步。

+1

“left”和“right”也是如此。 –

相关问题