给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
C++代码:
/**
* 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 * result = new ListNode(0);
ListNode * cur = result;
int x = 0, y = 0, carray = 0, total_val = 0, cur_val = 0;
while (l1 != nullptr || l2 != nullptr) // 只要有一个不为空,则继续
{
// 取值
x = (l1 != nullptr) ? l1->val : 0; // 有一个为空,则填0
y = (l2 != nullptr) ? l2->val : 0;
total_val = x + y + carray;
// 更新当前节点val和进位
cur_val = total_val % 10; // 取余数
carray = total_val / 10; // 取整数
// 插入新节点
cur->next = new ListNode(cur_val);
// 下一个节点
if (l1) l1 = l1->next; // 当前节点不空,则->next就不会报错
if (l2) l2 = l2->next;
cur = cur->next;
}
if (carray == 1)
cur->next = new ListNode(carray);
return result->next;
}
};
补充:要对头结点进行操作时,考虑创建哑节点dummy,使用dummy->next表示真正的头节点。这样可以避免处理头节点为空的边界问题。
python实现:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def addTwoNumbers(self, l1, l2):
result = cur = ListNode(0)
carray = 0
while l1 or l2: # 只要有一个不空,则继续
# 求和
x = l1.val if l1 else 0 # 如果为空,说明长度不够,用0
y = l2.val if l2 else 0
total_val = x + y + carray
# 更新当前节点值和进位数
cur_val = total_val % 10 # 余数
carray = total_val // 10 # 取整
# 插入当前节点
cur.next = ListNode(cur_val)
# 下一个
if l1: l1 = l1.next
if l2: l2 = l2.next
cur = cur.next
# l1和l2长度相同,且最后还有进位
if carray == 1: cur.next = ListNode(carray)
return result.next