LeetCode 算法 2.两数相加(python版)

需求

给你两个 非空 的链表,表示两个非负的整数。
它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

代码

# 定义链表节点类
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

def add_two_numbers(l1, l2):
    # 用于遍历链表的指针
    p1, p2 = l1, l2
    # 用于存储计算结果的链表的头节点和当前节点
    head = ListNode(0)
    current = head
    # 用于存储进位
    carry = 0

    while p1 or p2:
        # 获取当前节点的值,如果对应链表已经遍历完,则将值设为0
        x = p1.val if p1 else 0
        y = p2.val if p2 else 0
        # 计算和以及进位
        total = x + y + carry
        # 整除操作,取结果的整数部分
        carry = total // 10
        # 创建新节点并将其添加到结果链表中
        current.next = ListNode(total % 10)
        # 更新指针
        current = current.next
        if p1:
            p1 = p1.next
        if p2:
            p2 = p2.next

    # 处理最高位的进位
    if carry > 0:
        current.next = ListNode(carry)

    return head.next

# 创建第一个链表
l1 = ListNode(2)
l1.next = ListNode(4)
l1.next.next = ListNode(3)

# 创建第二个链表
l2 = ListNode(5)
l2.next = ListNode(6)
l2.next.next = ListNode(4)

# 调用函数得到结果链表
result = add_two_numbers(l1, l2)

# 遍历结果链表并打印每个节点的值
while result:
    print(result.val, end=",")
    result = result.next

运行结果

LeetCode 算法 2.两数相加(python版)_第1张图片

你可能感兴趣的:(算法,leetcode,python)