[力扣]2.两数相加 代码的优化

本文探讨了如何通过简化和优化代码,将原始的单链表加法问题转换为更高效、易读的形式。作者强调了追求高质量代码的重要性,并展示了原始代码与改进后的代码对比,通过合并while循环和逻辑简化,减少了冗余和复杂度。
摘要由CSDN通过智能技术生成

题目链接
这道题考察单链表操作的入门题目。
这道题并不难,写这篇博文并不是因为我不会,主要是想强调一下代码编写的问题。
—— 我们不仅要能写出代码,还应该追求高质量的代码!



野生代码

请看我最原始的代码。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        a = ListNode((l1.val + l2.val) % 10)
        p1,p2 = l1,l2
        p = a
        t = (l1.val + l2.val) // 10
        p1 = p1.next
        p2 = p2.next
        while p1 != None and p2 != None:
            t += p1.val + p2.val
            p.next = ListNode(t % 10)
            t //= 10
            p = p.next
            p1,p2 = p1.next,p2.next
        while p1 != None:
            t += p1.val
            p.next = ListNode(t % 10)
            t //= 10
            p = p.next
            p1 = p1.next
        while p2 != None:
            t += p2.val
            p.next = ListNode(t % 10)
            t //= 10
            p = p.next
            p2 = p2.next
        while t != 0:
            p.next = ListNode(t % 10)
            t //= 10
        return a

即不考虑代码的优化问题,完全按照最开始的思路来:

  1. t是用于进位的变量
  2. 对于l1l2的位数上都有值时,从低位到高位做相加计算
  3. 对只有l1位数上有值的计算
  4. 对只有l2位数上有值的计算
  5. 对进位不为0的计算

高质量代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        cur = dum = ListNode(-1)
        t = 0
        while l1 != None or l2 != None or t != 0:
            if l1 != None:
                t += l1.val
                l1 = l1.next
            if l2 != None:
                t += l2.val
                l2 = l2.next
            cur.next = ListNode(t % 10)
            cur = cur.next
            t //= 10
        return dum.next

原始代码中有多个while语句,通过在一个while语句中增加if判断,可以巧妙地将四个while合一。
实现代码的浓缩,避免代码冗长。


原创不易,感谢支持!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

wingaso

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值