[LeetCode]两数相加(Add Two Numbers)

119次阅读

共计 1436 个字符,预计需要花费 4 分钟才能阅读完成。

题目描述
给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。你可以假设除了数字 0 之外,这两个数字都不会以零开头。
示例:输入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 输出:7 -> 0 -> 8 原因:342 + 465 = 807
ListNode 数据结构
class ListNode {
int val;
ListNode next;

ListNode(int x) {
val = x;
}
}
解决方法
遍历链表对应相加即可,注意进位
方法一
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
// 为方便操作加入的头结点
ListNode res = new ListNode(0);
// 当前结点
ListNode cur = res;
// 进位标志
boolean[] flag = new boolean[1];

// 两链表对应位置相加
while (l1 != null && l2 != null) {
ListNode node = add(l1.val, l2.val, flag);
cur.next = node;
cur = node;

l1 = l1.next;
l2 = l2.next;
}

// 如果 l1 链表有剩余,则剩下的与 0 相加
while (l1 != null) {
ListNode node = add(l1.val, 0, flag);
cur.next = node;
cur = node;
l1 = l1.next;
}

// 如果 l2 链表有剩余,则剩下的与 0 相加
while (l2 != null) {
ListNode node = add(0, l2.val, flag);
cur.next = node;
cur = node;
l2 = l2.next;
}

// 如果链表末尾相加有进位则添加一个进位节点作为尾节点
if (flag[0])
cur.next = new ListNode(1);

return res.next;
}

public ListNode add(int v1, int v2, boolean[] flag) {
int sum = v1 + v2;
// 如果有进位则和 +1
if (flag[0]) {
sum++;
flag[0] = false;
}
// 产生进位
if (sum > 9)
flag[0] = true;
// 获取个位上的数
return new ListNode(sum % 10);
}
时间复杂度:O(max(M,N))。M,N 分别为 l1,l2 的链表长度
基于方法一的改进
public ListNode addTwoNumbers2(ListNode l1, ListNode l2) {
// 为方便操作加入的头节点
ListNode res = new ListNode(0);
// 当前节点
ListNode cur = res;
// 进位
int carry = 0;

// 两链表对应相加
while (l1 != null || l2 != null || carry != 0) {
int sum = (l1 != null ? l1.val : 0) + (l2 != null ? l2.val : 0) + carry;
// 获取进位
carry = sum / 10;
// 获取个位上的数
ListNode node = new ListNode(sum % 10);
cur.next = node;
cur = node;
l1 = l1 != null ? l1.next : null;
l2 = l2 != null ? l2.next : null;
}

return res.next;
}
时间复杂度:O(max(M,N))。M,N 分别为 l1,l2 的链表长度原文地址:https://lierabbit.cn/2018/04/…

正文完
 0