You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
思路:
看起来像是一道很简单的题,但是我却通过这道题发现了自己对指针掌握的不足。因为要返回一个链表,在链表的循环创建中指针是不断向下指的(ans),所以肯定要保留一个头结点(anshead)。就是在链表和链表头结点上,我纠结了很久。
首先,定义 anshead = new ListNode(0);
ans = anshead;
必须先给anshead分配内存,再把ans指向anshead分配的空间中,每次也必须先给 ans->next = new ListNode(0) 赋值后才能写 ans = ans->next;
总之,必须把ans 指向一个开辟过的空间。
否则, pre->next 在未开辟时指向 0x00000000 若这时 ans = pre->next 那么ans也是0x00000000 若这时给 ans 分配了空间,pre->next 还是0x00000000 因为之前ans没有明确的指定一个地方。
题目中只要注意进位就可以了。
class Solution { public: ListNode *addTwoNumbers(ListNode *l1, ListNode *l2) { if(l1 == NULL) return l2; if(l2 == NULL) return l1; ListNode * anshead = new ListNode(0); ListNode * ans = anshead; int cur = 0; //当前位的个位数 int up = 0; //进位 while(ans != NULL) //当没有下一位可以产生时就结束循环 { cur = up; if(l1 != NULL) { cur += l1->val; l1 = l1->next; } if(l2 != NULL) { cur += l2->val; l2 = l2->next; } up = cur / 10; cur = cur % 10; ans->val = cur; if(!(up == 0 && l1 == NULL && l2 == NULL)) //有进位 或 l1 l2 不空时才会有下一位 { ans->next = new ListNode(0); } ans = ans->next; } return anshead; } void createList(ListNode * &L) { int n; cin >> n; if(n != -1) { L = new ListNode(n); createList(L->next); } } };
看下大神同样思路,精简版的代码。
class Solution { public: ListNode *addTwoNumbers(ListNode *l1, ListNode *l2) { ListNode preHead(0), *p = &preHead; int extra = 0; while (l1 || l2 || extra) { int sum = (l1 ? l1->val : 0) + (l2 ? l2->val : 0) + extra; extra = sum / 10; p->next = new ListNode(sum % 10); p = p->next; l1 = l1 ? l1->next : l1; l2 = l2 ? l2->next : l2; } return preHead.next; //不要第一个自己随意赋的值,好方法 } };
时间: 2024-12-06 14:30:38