两数相加
给定两个非空链表,每个结点为非负整数,并且每个结点只存储数据元素只存储一位数字,按照逆序存储。将两个数相加,并以相同形式返回一个表示和的链表。
如:345 + 789 = 1134
代码实现
public class ListNode {
int val;
ListNode next;
ListNode() {}
ListNode(int val) { this.val = val; }
ListNode(int val, ListNode next) { this.val = val; this.next = next; }
}
class Solution {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode head = null;ListNode tail = null;
int carry = 0;
while(l1 != null || l2 != null){
int a1 = l1 != null ? l1.val : 0;
int a2 = l2 != null ? l2.val : 0;
int sum = a1 + a2 + carry;
if(head == null){
head = tail = new ListNode(sum % 10);
}else{
tail.next = new ListNode(sum % 10);
tail = tail.next;
}
carry = sum / 10;
if(l1 != null){
l1 = l1.next;
}
if(l2 != null){
l2 = l2.next;
}
}
if(carry > 0){
tail.next = new ListNode(carry);
}
return head;
}
}
本位和sum为当前位置上两个数的和加上当前位置进位值,即sum = a1 + a2 + carry,所求链表当前位置数字为sum % 10。初始时进位值carry为0,新的进位值为sum / 10。
同时遍历两个链表,若两个链表长度不同时,长度短的链表后对应位置为0。遍历结束后,若进位值carry大于0,需额外增加一个新节点,值为carry。
复杂度分析
- 时间复杂度:O(max(n1,n2)),其中n1,n2分别为两个链表的长度。遍历一遍链表所需时间。
- 空间复杂度:O(1)。