题目描述:
给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照逆序的方式存储的,并且它们的每个节点只能存储一位数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
我的代码:
一版
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode p, q;
ListNode C = new ListNode(0); // 创造节点C的头节点
ListNode s = C;
p = l1;
q = l2;
int c, f=0; // f用来做进位
while(p!=null&&q!=null){
c = p.val+q.val+f;
f = 0;
if (c>=10){
c = c-10;
f = 1;
}
s.next = new ListNode(c);
s = s.next;
p = p.next;
q = q.next;
}
// 如果l2比l1长
if (p==null&&q!=null) {
while (f == 1&&q!=null) {
c = q.val + f;
f=0;
if (c >= 10) {
c = c - 10;
f = 1;
}
s.next = new ListNode(c);
s = s.next;
s.next = q.next;
q = q.next;
}
if(q!=null){
s.next = q;
}
}
// 如果l1比l2长
if (q==null&&p!=null){
while (f==1&&p!=null){
c = p.val+f;
f = 0;
if (c>=10){
c = c-10;
f = 1;
}
s.next = new ListNode(c);
s = s.next;
s.next = p.next;
p = p.next;
}
if(p!=null){
s.next = p;
}
}
if(f==1){
s.next = new ListNode(f);
}
return C.next;
}
发现中间三部分代码相似读很高,导致代码冗长,做如下改进
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode p, q;
ListNode C = new ListNode(0); // 创造节点C的头节点
ListNode s = C;
p = l1;
q = l2;
int c, x, y;
int f=0;
while(p!=null||q!=null){
x = (p != null) ? p.val : 0;
y = (q != null) ? q.val : 0;
c = x+y+f;
f = 0;
if (c>=10){
c = c-10;
f = 1;
}
s.next = new ListNode(c);
s = s.next;
if(p!=null){p = p.next;}
if (q!=null){q = q.next;}
}
if(f==1){
s.next = new ListNode(f);
}
return C.next;
}
官方答案:
思想:
伪代码如下:
- 将当前结点初始化为返回列表的哑结点。
- 将进位 carry 初始化为 0。
- 将 p 和 q 分别初始化为列表 l1 和 l2 的头部。
- 遍历列表 l1 和 l2 直至到达它们的尾端。
- 将 x 设为结点pp 的值。如果 p 已经到达 l1 的末尾,则将其值设置为 0。
- 将 y 设为结点 q 的值。如果 q 已经到达 l2 的末尾,则将其值设置为 0。
- 设定 sum = x + y + carry。
- 更新进位的值,carry = sum / 10。
- 创建一个数值为 (sum mod 10) 的新结点,并将其设置为当前结点的下一个结点,然后将当前结点前进到下一个结点。
- 同时,将 p 和 q 前进到下一个结点。
- 检查 carry = 1是否成立,如果成立,则向返回列表追加一个含有数字 1 的新结点。
- 返回哑结点的下一个结点。
请注意,我们使用哑结点来简化代码。如果没有哑结点,则必须编写额外的条件语句来初始化表头的值。
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode dummyHead = new ListNode(0);
ListNode p = l1, q = l2, 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;
}