LeetCode之链表相加

最新推荐文章于 2024-08-12 11:10:27 发布

原创最新推荐文章于 2024-08-12 11:10:27 发布 · 3k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#编程 #字节跳动 #Leetcode #面试 #链表

编程专栏收录该内容

34 篇文章

订阅专栏

本文针对LeetCode题目AddTwoNumbersII，提出了一种通过翻转链表实现两数相加的方法。该方法首先将两个链表翻转，然后逐位相加并处理进位，最终返回新的链表结果。

博主在面试字节跳动实习生岗位时，遇见了链表求和的问题。而刷过 leetcode 的同学们对此也不会陌生，因为它就是 Leetcode 上的原题——Add Two Numbers II。因此，博主在此给出解题思路和解题方案，希望能对大家有所帮助。

简介

给定两个非空链表来代表两个非负整数。其中，数字最高位位于链表开始位置。它们的每个节点只存储单个数字。我们需要将这两数相加会返回一个新的链表。例如，(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4) = 7 -> 8 -> 0 -> 7

思路

经过分析，我们不难发现，根据加法原则，从低位开始相加，并且考虑进位标志。然而，链表开始位置是数字的最高位。因此，我们只需要先将链表进行翻转，然后根据加法原则即可完成此题。

代码

根据前一小节的分析，我们可有代码如下

public class ListNode {
    int val;
    ListNode next;
    ListNode(int x) { val = x; }
}

class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        if(l1 == null  || l2 == null) return null;

        // 栈保存数据
        Stack<Integer> s1 = new Stack<Integer>();
        Stack<Integer> s2 = new Stack<Integer>();

        ListNode p = l1;
        // 入栈
        while(p != null){
            s1.push(p.val);
            p = p.next;
        }

        p = l2;

        while(p != null){
            s2.push(p.val);
            p = p.next;
        }

        // 求和结果
        ListNode res = new ListNode(-1);
        p = res;
        int sum = 0;
        while(!s1.empty() || !s2.empty()){
            sum /= 10;
            if(!s1.empty()) sum += s1.pop();
            if(!s2.empty()) sum += s2.pop();

            ListNode newNode = new ListNode(sum % 10);
            newNode.next = p.next;
            p.next = newNode;
        }

        if(sum / 10 == 1){
            ListNode newNode = new ListNode(1);
            newNode.next = p.next;
            p.next = newNode;
        }
        return res.next;
    }