刷题打卡——链表求和

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

一、题目描述：

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

示例:

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912
复制代码

**进阶：**思考一下，假设这些数位是正向存放的，又该如何解决呢?

示例:

输入：(6 -> 1 -> 7) + (2 -> 9 -> 5)，即617 + 295
输出：9 -> 1 -> 2，即912
复制代码

二、思路分析：

根据题目分析，个位是在链表头部，所以只需要将节点值相加存到新的链表节点上，这里要注意相加进位问题，需要用一个临时变量存储进位。

当链表1和链表2其中一个节点不为null时就需要计算和，如果其中一个为null而另一个有值时null作为0相加。

进阶的话，可以使用两个栈，利用栈的后进先出，先把两个链表存到栈中，然后依次取出计算。注意和节点的顺序。重点在于 a.next = b.next ，b.next = a 这个操作，将节点反转。

三、AC 代码：

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        int temp = 0; //进位
        ListNode sum = new ListNode(0); //链表和
        ListNode s = sum;
        while(l1 != null || l2!=null){
            int n = 0;
            if(l1 != null){
                n += l1.val;
                l1 = l1.next;
            }
            if(l2 != null){
                n += l2.val;
                l2 = l2.next;
            }
            n += temp;
            ListNode node = new ListNode(n%10);
            s.next = node;
            s = s.next;
            temp = n/10; //进位
        }
        if(temp>0){
            s.next = new ListNode(temp);
        }
        return sum.next;
    }
}
复制代码

进阶

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        int temp = 0; //进位
        ListNode sum = new ListNode(0); //链表和
        Stack<ListNode> s1 = new Stack<>();
        Stack<ListNode> s2 = new Stack<>();

        while(l1!=null){
            s1.push(l1);
            l1 = l1.next;
        }
        while(l2!=null){
            s2.push(l2);
            l2 = l2.next;
        }

        while(!s1.isEmpty() || !s2.isEmpty()){
            int n = 0;
            if(!s1.isEmpty())
                n += s1.pop().val;
            if(!s2.isEmpty())
                n += s2.pop().val;

            n += temp;
            ListNode node = new ListNode(n%10);
            temp = n/10;
            node.next = sum.next;
            sum.next = node;

        }
        return sum.next;
    }
}
复制代码

总结

本题主要练习的是链表的新增操作，同时也用到了栈的后进先出特性，坚持刷题第15天，快来和我一起学习打卡吧。