两数相加
这是一道来自leetcode的算法题目。
题目描述
给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照逆序的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
解题思路
和基础的四则运算的思路一样,由于是逆序方式存储的单数字,从头节点同时开始遍历链表,即从个位开始相加,还要注意进位问题,可以用一个标记整数变量carry,carry为0即不进位。
算法
- 创建结果头节点,创建两个新指针p,q分别指向已知链表;
- 初始化标记进位的变量carry为0;
- 判断p,q是否不全为空,是则进入循环,遍历链表,即进行第4步,否则跳出循环,即进行第8步;
- 创建两个变量x,y,如果当前节点为空(null)则变量赋为0,否则赋予当前节点的值(val);
- 创建一个临时变量sum,sum=x+y+carry;
- 判断是否需要进位,实际过程中carry=sum/10即可解决,不需判断是否大于10;
- 指向下一节点,进行第3步;
- 判断是否需要再进位;
- 返回结果;
代码
Java
/**
* Definition for singly-linked list.
* */
public class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
class Solution {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode result = new ListNode(0);
ListNode p = l1,q = l2, curr = result;
int carry = 0;
while(p!=null||q!=null){
int x = p == null ? 0 : p.val;
int y = q == null ? 0 : q.val;
int sum = x + y + carry;
carry = sum/10;
sum = sum%10;
curr.next = new ListNode(sum);
curr = curr.next;
if(p != null) p = p.next;
if(q != null) q = q.next;
}
if(carry>0){
curr.next = new ListNode(carry);
}
return result.next;
}
}
python3
# Definition for singly-linked list.
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
dummyHead = ListNode(0)
curr, carry = dummyHead, 0
while l1 or l2:
sum = 0
if l1:
sum += l1.val
l1 = l1.next
if l2:
sum += l2.val
l2 = l2.next
sum += carry
carry = sum // 10
curr.next = ListNode(sum % 10)
curr = curr.next
if carry > 0:
curr.next = ListNode(1)
return dummyHead.next;