算法题之两个链表相加(中等难度)
思路
⾸先每⼀位相加肯定会产⽣进位,我们⽤ carry 表示。进位最⼤会是 1 ,因为最⼤的情况是⽆⾮是 9 + 9 +
1 = 19 ,也就是两个最⼤的数相加,再加进位,这样最⼤是 19 ,不会产⽣进位 2 。下边是伪代码。
初始化⼀个节点的头,result head ,但是这个头不存储数字。并且将 curr 指向它。
初始化进位 carry 为 0 。
初始化 p 和 q 分别为给定的两个链表 l1 和 l2 的头,也就是个位。
循环,直到 l1 和 l2 全部到达 null 。
设置 x 为 p 节点的值,如果 p 已经到达了 null,设置 x 为 0 。
设置 y 为 q 节点的值,如果 q 已经到达了 null,设置 y 为 0 。
设置 sum = x + y + carry 。
更新 carry = sum / 10 。
创建⼀个值为 sum mod 10 的节点,并将 curr 的 next 指向它,同时 curr 指向变为当前的新节点。
向前移动 p 和 q 。
若到达最后,还需要判断 carry 是否等于 1 ,如果等于 1 (则表示刚才进位了),所以还需要在链表末尾增加⼀个为 1 的节点。
返回 result head的 next ,也就是个位数开始的地⽅。
初始化的节点 result head 没有存储值,最后返回 result head 的 next 。这样的好处是不⽤单独对
head 进⾏判断改变值。也就是如果⼀开始的 head 就是代表个位数,那么开始初始化的时候并不知道它的
值是多少,所以还需要在进⼊循环前单独对它进⾏值的更正,不能像现在⼀样只⽤⼀个循环简洁。
图解(图画的不好,请谅解)
public class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
/**
* 两个链表相加
*/
public class TwoListNodeAdd {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
//结果的头节点,只是为了下面不用去判断是否有头节点
ListNode resultHead = new ListNode(0);
ListNode p = l1;
ListNode q = l2;
ListNode cur = resultHead;
int carry = 0;//这个变量就是判断前两个节点是否进位了,初始值为0
while (p != null || q != null){
int x = (p != null) ? p.val : 0;
int y = (q != null) ? q.val : 0;
int sum = x + y +carry;
carry = sum /10;//这个操作就是判断当前的两个节点的值是否大于10,若大于10就进位
cur.next = new ListNode(sum%10);//这个就是把当前两个节点的和创建一个新节点
q = (q != null ? q.next: null);
p = (p != null ? p.next : null);
}
//出循环以后要判断carry当前的值,如不为0,说明
if (carry != 0){
cur.next = new ListNode(1);
}
return resultHead.next;
}
}
- 时间复杂度:O(max(m,n)),m 和 n 代表 l1 和 l2 的⻓度。
- 空间复杂度:O(max(m,n)),m 和 n 代表 l1 和 l2 的⻓度。⽽其实新的 List 最⼤⻓度是 O(max(m,n))+ 1,因为我们的 head 没有存储值。