这段时间开始在leetcode上面做题,然后遇到了一个很奇怪的问题。
下面这一段是可以正确执行的代码
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
struct ListNode* root = (struct ListNode*)malloc(sizeof(struct ListNode));
struct ListNode* p = root;
struct ListNode* temp;
int carry = 0;
int mod;
while(l1||l2||carry)
{
temp = l1?l1:l2;
if(l1)
{
carry += l1->val;
l1 = l1->next;
}
if(l2)
{
carry += l2->val;
l2 = l2->next;
}
mod = carry%10;
carry = carry/10;
if(temp)
{
temp->val = mod;
p = p->next = temp;
}else
{
struct ListNode* nNode = new struct ListNode(mod);//接下来将要修改这一部分。
nNode->val = mod;
nNode->next = NULL;
p->next = nNode;
p = p->next;
}
}
return root->next;
}
};
然后,我把分配新结点的方式改成了声明变量的方式
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
struct ListNode* root = (struct ListNode*)malloc(sizeof(struct ListNode));
struct ListNode* p = root;
struct ListNode* temp;
int carry = 0;
int mod;
while(l1||l2||carry)
{
temp = l1?l1:l2;
if(l1)
{
carry += l1->val;
l1 = l1->next;
}
if(l2)
{
carry += l2->val;
l2 = l2->next;
}
mod = carry%10;
carry = carry/10;
if(temp)
{
temp->val = mod;
p = p->next = temp;
}else
{
struct ListNode nNode(mod);//使用声明变量的方式创建新结点nNode,保存最后的一个进位。
#nNode->val = mod;
#nNode->next = NULL;
p->next = nNode;
p = p->next;
}
}
return root->next;
}
};
这样的代码如果都是写到主函数里面肯定是没问题的。但是,在这里类里面,得到的root->next的确也是正确的结果。只是这里声明的结构体变量nNode等到程序回到主函数里面的时候就会被回收掉了,然后就遇到了不会有结果输出的error。
补充一些内存分配的知识。
1. C/C++程序中,分配的内存一般在堆和栈中。
分配在堆区中的内存,一般由程序员手动回收,比如说free、delete。或者程序结束,系统自动回收。
分配在栈区中的内存,在该函数声明周期结束之后就被收掉了。就像我上面写的这个函数,虽然在addTwoNumbers函数中程序能够正确返回指向结果链表的头指针,但是返回到主程序之后,该指针所指向的内存就被回收了。。。
2. new和malloc
new和malloc分配的内存在堆上,也称为动态内存。
-------------------------------------------------------------------------------------------------------------------------
参考: http://blog.csdn.net/jing0611/article/details/4030237
-------------------------------------------------------------------------------------------------------------------------
Good good study, day day up~~