题目描述
题目大致的意思是给输入是两个非负的数字,数字由链表存储,一个节点存储一个数,比如342存储为2 -> 4 -> 3。
在题目中所给的测试样例
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.
思路
这道题和在国内经常做的大数加法由异曲同工之妙,我记得本科那会的竞赛书上会出这样的一道题,只不过那道题容器是数组,这道题容器是链表。
考虑到容器是链表,就需要考虑链表是否为空,以及next指针的操作。比如两个数字位数不相同情况的处理,以及进位的处理。
链表循环
设两个数字的链表为L1和L2,在进行循环加法操作时,只要L1和L2任意一位不为空,循环则需继续。
while(l1 || l2){
……
l1 = l1? l1->next : NULL;
l2 = l2? l2->next : NULL;
}
加数处理
这里有若干个小细节,下面来分别解析。我们遇到的情况有以下三种:
- L1、L2都不为空
- L1为空、L2不为空
- L1不为空、L2为空
最“笨“的方法其实可以用三个if来实现三种情况,但是我们可以想着将这三种情况合成一种情况处理,为空的情况就是为零,为空的话就当0做加法。
int x1 = l1 ? l1->val : 0;
int x2 = l2 ? l2->val : 0;
求和处理
这里就考虑到前面一位的进位,初始化时赋值为0,每次求和运算算出是否有进位
int carry = 0;
int temp = x1 + x2 + carry;
carry = temp / 10;
int val = temp % 10;
建立和的链表
我们需要建立两个链表的指针,一个来存储头节点的地址,一个来进行新节点的生成。在这里需要重新温习一下结构体的初始化。
ListNode* head = NULL;//头节点
ListNode* p =NULL;//实时更新的节点
ListNode* newPointer = new ListNode(val);
//C++中Struct可以通过new来初始化
对于一开始,需要将newPointer赋值于head和p,之后的循环只对p进行操作即可。
if(!head){
head = newPointer;
}
if(p){//表示p不是第一次赋值
p->next = newPointer;
}
p = newPointer;
最后的进位
两个数的和最后可能会比两个数的位数多处一位(考虑加和进1),所以要单独将最后的carry生成一个新的节点。
if(carry){
ListNode* newPointer = new ListNode(carry);
p->next = newPointer;
}
源码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = NULL;
ListNode* p = NULL;
int carry = 0;
while(l1 || l2){
int x1 = l1 ? l1->val : 0;
int x2 = l2 ? l2->val : 0;
int temp = x1 + x2 + carry;
int val = temp % 10;
carry = temp / 10;
ListNode* newPointer = new ListNode(val);
if(!head){
head = newPointer;
}
if(p){
p->next = newPointer;
}
p = newPointer;
l1 = l1 ? l1->next : NULL;
l2 = l2 ? l2->next : NULL;
}
if(carry){
ListNode* newPointer = new ListNode(carry);
p->next = newPointer;
}
return head;
}
};