1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
| class Solution { public: ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) { bool note = false; ListNode* head = new ListNode(-1); ListNode* rehead = head; while (l1 != nullptr && l2 != nullptr) { int num = l1->val + l2->val; if (note) { num++; note = false; } if (num >= 10) { num = num % 10; note = true; } ListNode* tmp = new ListNode(num); rehead->next = tmp; rehead = rehead->next; l1 = l1->next; l2 = l2->next; } if (l2) { swap(l1, l2); } while (l1 != nullptr) { int num = l1->val; if (note) { num++; note = false; } if (num >= 10) { num = num % 10; note = true; } ListNode* tmp = new ListNode(num); rehead->next = tmp; rehead = rehead->next; l1 = l1->next; } if (note) { ListNode* tmp = new ListNode(1); rehead->next = tmp; } return head->next; } };
|
还是要看懂题目,实际上就是将两个链表从头到尾取相同位置的节点相加,这可能出现以下情况:
- 两个链表长度相同且最后没有产生进位,返回 head->next 即可
- 两个链表长度相同且最后有产生进位,创建一个值为 1 的新节点 tmp,然后 rehead->next 指向 tmp节点,最后返回 head->next 即可
- 两个链表长度不同,那么对于没有处理完的链表中的节点逐一移动到 rehead 节点之后,接着进入上述的两种情况中继续处理
1 2 3
| 输入:l1 = [5,6,4,9], l2 = [2,4,9] 输出:[7,0,4,0,1] 解释:9465 + 942 = 10407
|
当然,还有就是关于进位的处理,这并不难,只需要用一个标记位 note 来标识是否有进位,还有要记得正确更新 note 的情况。
1 2 3 4 5 6 7 8
| if (note) { num++; note = false; } if (num >= 10) { num = num % 10; note = true; }
|