2. 两数相加
题目描述
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
思路
就像你在纸上计算两个数字的和那样,我们首先从最低有效位也就是列表 l1 和 l2 的表头开始相加。由于每位数字都应当处于0…9 的范围内,我们计算两个数字的和时可能会出现 “溢出”。例如,5 + 7 = 12。在这种情况下,我们会将当前位的数值设置为 22,并将进位 carry = 1带入下一次迭代。进位 carry 必定是 00 或 11,这是因为两个数字相加(考虑到进位)可能出现的最大和为 9+9+1=19。
伪代码如下:
1.将当前结点初始化为返回列表的哑结点。
2.将进位 carry初始化为 0。
3.将 p 和 q 分别初始化为列表 l1和 l2的头部。
4.遍历列表 l1和 l2直至到达它们的尾端。
将 x 设为结点 p 的值。如果 p 已经到达 l1 的末尾,则将其值设置为 0。
将 y 设为结点 q 的值。如果 q 已经到达 l2 的末尾,则将其值设置为 0。
设定 sum = x + y + carry。
更新进位的值,carry = sum / 10。
创建一个数值为 (sum mod 10) 的新结点,并将其设置为当前结点的下一个结点,然后将当前结点前进到下一个结点。
同时,将 p 和 q 前进到下一个结点。
5.检查 carry = 1是否成立,如果成立,则向返回列表追加一个含有数字 1 的新结点。
6.返回哑结点的下一个结点。
代码
/**
* 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* dummyHead = new ListNode(0);
ListNode* p = l1;
ListNode* q = l2;
ListNode* curr = dummyHead;
int carry = 0;
while(p != NULL || q != NULL){
int x = (p != NULL) ? p->val : 0;
int y = (q != NULL) ? q->val : 0;
int sum = carry + x + y;
carry = sum / 10;
curr->next = new ListNode(sum % 10);
curr = curr->next;
if(p != NULL){
p = p->next;
}
if(q != NULL){
q = q->next;
}
}
if(carry > 0){
curr->next = new ListNode(carry);
}
return dummyHead->next;
}
};
复杂度分析
时间复杂度:O(max(m,n)),假设 m 和 n 分别表示 l1 和 l2 的长度,上面的算法最多重复 max(m,n) 次。
空间复杂度:O(max(m,n)), 新列表的长度最多为 max(m,n)+1。