[链表遍历|模拟] leetcode 2 两数相加
1.题目
题目链接
给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照逆序的方式存储的,并且它们的每个节点只能存储一位数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
2.分析
2.1.题意理解
本题模拟的是十进制加法的过程。由于我们人工进行加法是从最低位(个位)开始的,因此题目中个位为链表头结点,以方便我们进行遍历。
也就是说,我们要计算a与b相加的和c。c的第i位c[i]=a[i]+b[i]+carry。其中carry表示进位。
2.2.对齐问题
相加的两个数的长度并不一定相同。例如123+45,实际上我们手工计算的时候给45补上了0,即123+045。在链表中模拟补0的过程即为对齐的过程:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
while(l1 != NULL || l2 != NULL) {
//如果遍历完链表尾则跳过,相当于c[i] += 0
if(l1 != NULL) {
//do something...
}
if(l2 != NULL) {
//do something...
}
}
}
2.3.进位问题
可以用一个变量carry来模拟两数相加过程中产生的进位。由于是两个数做加法,因此进位只可能为1或0,可以用布尔值表示:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
while(l1 != NULL || l2 != NULL) {
//sum即为之前提到的c[i]
if(carry) {
sum++; // c[i] = a[i] + b[i] + carry
}
carry = sum>=10?true:false; //逢十进位
}
}
3.代码
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = new ListNode(-1); //存放结果的链表
ListNode* h = head; //用于将c[i]写入结果链表的指针
int sum = 0; //c[i]
bool carry = false; //进位标志carry
while(l1 != NULL||l2 != NULL) { //遍历两个链表
sum = 0; //c[i] = 0
if(l1 != NULL) {
sum += l1->val; //c[i] += a[i]
l1 = l1->next; //继续遍历
}
if(l2 != NULL) {
sum += l2->val; //c[i] += b[i]
l2 = l2->next;
}
if(carry) {
sum++; //c[i] += carry
}
//将c[i]写入链表结点
h->next = new ListNode(sum % 10);
h = h->next;
//更新carry
carry = sum>=10?true:false;
}
//每一位均相加完毕后还可能剩余进位
if(carry) {
h->next = new ListNode(1);
}
return head->next;
}
};