题目描述
You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
解题思路
这道题目很简单,链表表示的是从低位到高位的逆序的数字,很符合我们做加法的习惯。需要注意的是十进制相加时,逢十进一的情况。本题和Plus One的思路类似。
我们可以使用变量
int carry;
表示需要进的位数。例如在上面的例子中,
carry = 0;
2 + 5 + carry = 7 -> carry = 0;
4 + 6 + carry = 10 -> carry = (4 + 6) / 10; -> carry = 1;
3 + 4 + carry = 8 -> carry = 0;
每次相加的时候加上carry的值。
若遍历到链表结尾,而
carry != 0 is true
那么我们需要在链表的结尾加上一位,表示新的最高位。
类似的当一个链表达到最后一个结点,另一个未到最后一个结点时,我们需要结合此时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) {
if (!l1 && !l2)
return NULL;
if (l1 && !l2)
return l1;
if (!l1 && l2)
return l2;
ListNode* h;
ListNode* l = new ListNode((l1->val+l2->val)%10);
int carry = (l1->val + l2->val) / 10;
h = l;
l1 = l1->next;
l2 = l2->next;
while (l1 && l2)
{
ListNode* n = new ListNode((l1->val+l2->val+carry)%10);
carry = (l1->val + l2->val + carry) / 10;
l->next = n;
l = l->next;
l1 = l1->next;
l2 = l2->next;
}
if(!carry && l1)
l->next = l1;
else if (!carry && l2)
l->next = l2;
else if (carry && l1)
{
while(l1)
{
ListNode* n = new ListNode((l1->val+carry)%10);
carry = (l1->val+carry) / 10;
l->next = n;
l = l->next;
l1 = l1->next;
if (!carry)
{
l->next = l1;
break;
}
}
}
else if (carry && l2)
{
while(l2)
{
ListNode* n = new ListNode((l2->val+carry)%10);
carry = (l2->val+carry) / 10;
l->next = n;
l = l->next;
l2 = l2->next;
if (!carry)
{
l->next = l2;
break;
}
}
}
if (carry)
l->next = new ListNode(1);
return h;
}
};