刷题-力扣-面试题 02.05. 链表求和
题目链接
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-lists-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目描述
给定两个用链表表示的整数,每个节点包含一个数位。
这些数位是反向存放的,也就是个位排在链表首部。
编写函数对这两个整数求和,并用链表形式返回结果。
示例:
输入:(7 -> 1 -> 6) + (5 -> 9 -> 2),即617 + 295
输出:2 -> 1 -> 9,即912
进阶:思考一下,假设这些数位是正向存放的,又该如何解决呢?
示例:
输入:(6 -> 1 -> 7) + (2 -> 9 -> 5),即617 + 295
输出:9 -> 1 -> 2,即912
题目分析
- 根据题目描述,求两个用链表表示的整数到的和
- 从低位向高位求和,使用carry表示进位
代码
/**
* 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* res = new ListNode();
ListNode* p = res;
int carry = 0;
while (l1 || l2) {
if (l1 && l2) {
int sum = carry + l1->val + l2->val;
carry = sum / 10;
p->next = new ListNode(sum % 10);
l1 = l1->next;
l2 = l2->next;
p = p->next;
} else if (l1) {
if (carry == 1 && l1->val == 9) {
p->next = new ListNode(0);
} else {
p->next = new ListNode(carry + l1->val);
carry = 0;
}
l1 = l1->next;
p = p->next;
} else if (l2) {
// l2不存在
if (carry == 1 && l2->val == 9) {
p->next = new ListNode(0);
} else {
p->next = new ListNode(carry + l2->val);
carry = 0;
}
l2 = l2->next;
p = p->next;
}
}
if (carry) {
p->next = new ListNode(1);
}
p = res->next;
delete res;
return p;
}
};