题
面试题 02.05. 链表求和
给定两个用链表表示的整数,每个节点包含一个数位。
这些数位是反向存放的,也就是个位排在链表首部。
编写函数对这两个整数求和,并用链表形式返回结果。
示例:
输入:(7 -> 1 -> 6) + (5 -> 9 -> 2),即617 + 295
输出:2 -> 1 -> 9,即912
进阶:思考一下,假设这些数位是正向存放的,又该如何解决呢?
示例:
输入:(6 -> 1 -> 7) + (2 -> 9 -> 5),即617 + 295
输出:9 -> 1 -> 2,即912
代码
- 可以使用创建新的列表 推荐这种写法
- 也可以使用l1 l2作为存储 但是时间稍微复杂 代码量增大 下面代码就是这一种 代码量很大细节太多。。但是内存占用很小
/**
* 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) {
//这个数不可能>9
short int needAdd = 0;
ListNode *ans = new ListNode(0), *p = ans, *list2 = l2, *temp;
int model = 1;
while (l1 || l2)
{
int value = needAdd;
if (l1)
{
value += l1->val;
}
if (l2)
{
value += l2->val;
}
needAdd = value / 10;
value %= 10;
if (model == 1) {
l1->val = value;
p->next = l1;
temp = l1->next;
l1->next = NULL;
l1 = temp;
if (l2)
{
l2 = l2->next;
}
}
else if (model == 2)
{
list2->val = value;
p->next = list2;
list2 = list2->next;
p->next->next = NULL;
l2 = l2->next;
}
if (l1 == NULL)
{
model = 2;
}
p = p->next;
}
if (needAdd > 0)
{
p->next = new ListNode(needAdd);
}
return ans->next;
}
};