2016-09-18 64 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」也是如此。 –

相關問題