【c++】返回类的指针需要注意的问题

问题是在刷剑指 offer时发现的

具体问题如下

定义了类TreeNode

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

递归实现由先序遍历和中序遍历构造二叉树

问题出在构建类对象和返回指向类的指针上 

line6 和 最后的return，构造过程中会出现内存溢出，访问到不合法位置

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size() == 0)
            return NULL;
        else if(pre.size() == 1)
            return new TreeNode(pre[0]);
        TreeNode root(pre[0]);
        int f;
        for(int i = 0; i < vin.size(); ++i){
            if(vin[i] == pre[0]){
                f = i;
                break;
            }
        }
        root.left = reConstructBinaryTree(vector<int>(pre.begin() + 1, pre.begin() + f + 1),
                                          vector<int>(vin.begin(), vin.begin() + f));
        root.right = reConstructBinaryTree(vector<int>(pre.begin() + f + 1, pre.end()),
                                          vector<int>(vin.begin() + f + 1, vin.end()));
        return &root;
}

修改对象的构造方法后，正确运行

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size() == 0)
            return NULL;
        else if(pre.size() == 1)
            return new TreeNode(pre[0]);
        TreeNode* root = new TreeNode(pre[0]);
        int f;
        for(int i = 0; i < vin.size(); ++i){
            if(vin[i] == pre[0]){
                f = i;
                break;
            }
        }
        root->left = reConstructBinaryTree(vector<int>(pre.begin() + 1, pre.begin() + f + 1),
                                          vector<int>(vin.begin(), vin.begin() + f));
        root->right = reConstructBinaryTree(vector<int>(pre.begin() + f + 1, pre.end()),
                                          vector<int>(vin.begin() + f + 1, vin.end()));
        return root;
    }
};

将构造方法变为new一个返回指向对象指针的 TreeNode* 对象，修改访问部分的代码，正确运行

第一种方法造成错误的原因是因为TreeNode root(pre[0]);构造的root对象分配的栈类型的内存，在函数结束后就会释放内存，这时仍然返回指向这个栈的地址，但是这个栈中的数据已经释放，会造成指向不合法的对象。

第二种方法，new一个对象，分配的是堆类型的内存

从堆上分配，亦称动态内存分配。程序在运行的时候用malloc或new申请任意多少的内存，程序员自己负责在何时用free或delete释放内存。动态内存的生存期由我们决定，使用非常灵活，但问题也最多。

在我们delete释放内存前，会一直在程序运行过程中存在，所以是合法的。