LeetCode 02:“两数相加”,小学加法运算而已?

心已赠人 2023-01-07 10:08 227阅读 0赞

一道小学加法题,竟然在LeetCode上被标记为“中等”难度,有些人“流下了没有技术的眼泪”,有些人“一顿操作猛如虎,一看击败百分五……”。今天我们来看看LeetCode的第二道题“两数相加”。

“两数相加”

先来看题目描述,对应官方链接:https://leetcode-cn.com/problems/add-two-numbers

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

链表节点的数据结构如下:

  1. public class ListNode {
  2. int val;
  3. ListNode next;
  4. ListNode() {}
  5. ListNode(int val) { this.val = val; }
  6. ListNode(int val, ListNode next) { this.val = val; this.next = next; }
  7. }

题目说明

题目描述相对来说比较绕,我们可以直接理解为两个多位的整数相加,只不过整数的每一位都是通过链表进行存储。比如,整数342,通过链表存储正常来说应该是3->4->2,但是计算时,往往需要从低位开始计算,逢十进一,所以题目中直接将整数表示为2->4->3,这样反而不用将链表顺序进行反转了,直接相加就可以了。
在这里插入图片描述

需要注意的是如果两个链表的长度不同,则可以认为长度短的链表的后面有若干个 0 ,链表遍历结束,则如果进位值大于0,则还需要在结果链表中附加一个值为1的节点。

方法一:模拟

上面已经提到,链表是逆序的,因此直接对应数字相加即可。基本操作遍历两个列表,逐位计算它们的和,并与当前位置的进位值相加。

比如,两个链表对应位的数字分别为n1和n2,进位为carry(通常为0和1),则它们的和为(n1 + n2 + carry),对应位上数字变为(n1 + n2 + carry)%10,新的进位为(n1 + n2 + carry)/10。

如果两个链表长度不一样,长度短的链表后续对应位上值均为0即可。如果遍历结束之后,carray大于0(也就是等于1),则在结构链表后面新增一个节点,

实现代码如下:

  1. class Solution {
  2. public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
  3. ListNode head = null, tail = null;
  4. int carry = 0;
  5. while (l1 != null || l2 != null) {
  6. int n1 = l1 != null ? l1.val : 0;
  7. int n2 = l2 != null ? l2.val : 0;
  8. int sum = n1 + n2 + carry;
  9. if (head == null) {
  10. head = tail = new ListNode(sum % 10);
  11. } else {
  12. tail.next = new ListNode(sum % 10);
  13. tail = tail.next;
  14. }
  15. carry = sum / 10;
  16. if (l1 != null) {
  17. l1 = l1.next;
  18. }
  19. if (l2 != null) {
  20. l2 = l2.next;
  21. }
  22. }
  23. if (carry > 0) {
  24. tail.next = new ListNode(carry);
  25. }
  26. return head;
  27. }
  28. }

上述方法时间复杂度的计算与链表的长度有关,比如两个链表的长度分别为m和n,则遍历的次数为max(m,n),也就m和n中取最大值,所以时间复杂度为O(n)。

由于要对链表的每一位进行计算存储,并且最后如果有进位,还要多加一位,因此最长链表为max(m,n)+1,所以空间复杂度为O(n);

通过思路分析,写出上面的代码还是比较容易的。但这个题目是否可以考虑用递归的形式来解决呢?我们来看看方法二。

方法二:递归

第一种方法很简单,按照正常的思维逻辑来就可以了。但评论区有这样一句话“不用递归没有灵魂。尽管多数时候,递归不见得更有效率。”那么我们就来看看用递归的形式如何实现。

  1. class Solution {
  2. public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
  3. return add(l1,l2,0);
  4. }
  5. public ListNode add(ListNode l1, ListNode l2, int carry){
  6. if(l1 == null && l2 == null && carry == 0) return null;
  7. int x = l1==null ? 0 : l1.val;
  8. int y = l2==null ? 0 : l2.val;
  9. int sum = x + y + carry;
  10. ListNode n = new ListNode(sum % 10);
  11. n.next = add(l1==null ? null : l1.next,
  12. l2==null ? null : l2.next,
  13. sum/10);
  14. return n;
  15. }
  16. }

上述代码的基本迭代逻辑循环如下:
在这里插入图片描述

通过上图我们可以推演一下递归调用的时间复杂度。针对递归调用的时间复杂度计算,本质上要看:递归的次数✖️每次递归中的操作次数。那么,上述方法递归了几次呢?递归的次数也是与两个链表最长的那个的长度有关,最后可能会因为进位多算一次,因此递归次数为n或n+1,而内部计算并不随n的变化而变化,执行次数为常数。因此整体时间复杂度为n*1 = O(n)。

空间复杂度依旧与结果链表的长度有关,因此依旧为O(n)。

小结

就算法本身而言是比较简单的,理清思路,逐渐添加判断即可获得解法。重点在于大家是否能够想到通过递归算法来进行解答。本道题递归算法并没有让时间复杂度降低,而在某些情况下通过递归算法能将时间复杂度从O(n)降低到O(logn),这将是很大性能提升。比如,求x的n次方,大家可以尝试一下。

原文链接:《LeetCode 02:“两数相加”,小学加法运算而已?》


程序新视界
公众号“ 程序新视界”,一个让你软实力、硬技术同步提升的平台,提供海量资料

微信公众号:程序新视界

发表评论

表情:
评论列表 (有 0 条评论,227人围观)

还没有评论,来说两句吧...

相关阅读

    相关 leetcode 相加

    给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。 如果,我们将这两个数相加起来,则会返回

    相关 LeetCode(相加)

    题目 给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。 如果,我们将这两个数相加起来