leetcode 链表专题 445——两数相加
给你两个 非空 链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储一位数字。将这两数相加会返回一个新的链表。你可以假设除了数字 0 之外,这两个数字都不会以零开头。
进阶:如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。
示例:
输入:(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 8 -> 0 -> 7
思路:先把两个链表反转,然后相加得到一个链表,最后将得到的链表反转。
注意:如果循环条件,如果链表1和链表2的值都为空,但是上一次相加有进位,也需要继续进行。例子:链表1【5】,链表2【5】,相加之和为【0,1】,反转之后为【1,0】
package algorithm_linkedList;
public class Solution445 {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode newl1 = reverse(l1);
ListNode newl2 = reverse(l2);
ListNode newHead = null, next = null;
int mod = 0;
while(newl1 != null || newl2 != null || mod != 0) {
int val1 = newl1 != null ? newl1.val : 0;
int val2 = newl2 != null ? newl2.val : 0;
int sum = val1 +val2 + mod;
ListNode node = new ListNode(sum%10);
if(newHead == null) {
newHead = next = node;
}else {
next.next = node;
next = next.next;
}
mod = sum / 10;
if(newl1 != null) newl1 = newl1.next;
if(newl2 != null) newl2 = newl2.next;
}
return reverse(newHead);
}
// 链表反转
public ListNode reverse(ListNode head) {
if(head == null || head.next ==null)
return head;
ListNode node = head.next;
ListNode newHead = reverse(node);
node.next = head;
head.next = null;
return newHead;
}
// 打印链表
public void printList(ListNode head) {
while(head != null) {
System.out.print(head.val + " ");
head = head.next;
}
System.out.println();
}
public static void main(String[] args) {
ListNode node1 = new ListNode(5);
ListNode node2 = new ListNode(7);
ListNode node3 = new ListNode(4);
ListNode node4 = new ListNode(5);
ListNode node5 = new ListNode(3);
node1.next = node2;
node2.next = node3;
node4.next = node5;
Solution445 test = new Solution445();
test.printList(node1);
test.printList(node4);
ListNode add = test.addTwoNumbers(node1, node4);
test.printList(add);
}
}